TT-MLIR
|
Enumerations | |
enum class | EltwiseOpType : uint32_t { Add = 0 , Multiply = 1 , Subtract = 2 , Relu = 3 , GreaterEqual = 4 , Sqrt = 5 , Div = 6 , Sigmoid = 7 , Reciprocal = 8 , Exp = 9 , Maximum = 10 , Abs = 11 , Neg = 12 , Rsqrt = 13 , Typecast = 14 , Equal = 15 , NotEqual = 16 , LessEqual = 17 , LessThan = 18 , GreaterThan = 19 , LogicalAnd = 20 , LogicalOr = 21 , LogicalXor = 22 , LogicalNot = 23 , BitwiseAnd = 24 , BitwiseOr = 25 , BitwiseXor = 26 , BitwiseNot = 27 , Cbrt = 28 , Minimum = 29 , Ceil = 30 , Sin = 31 , Cos = 32 , Log = 33 , Log1p = 34 , Expm1 = 35 , Sign = 36 , Remainder = 37 , IsFinite = 38 , Floor = 39 , Where = 40 , Gelu = 41 , Clamp = 42 , LeakyRelu = 43 , Scatter = 44 , Tan = 45 , Tanh = 46 , MIN = Add , MAX = Tanh } |
enum class | EltwiseOpParams : uint8_t { NONE = 0 , ClampOpParams = 1 , EltwiseOpWithFloatParams = 2 , MIN = NONE , MAX = EltwiseOpWithFloatParams } |
enum class | ReductionOpType : uint32_t { Sum = 0 , Mean = 1 , Max = 2 , MIN = Sum , MAX = Max } |
enum class | OpType : uint8_t { NONE = 0 , GetDeviceOp = 1 , ToMemoryConfigOp = 2 , ToLayoutOp = 3 , TypecastOp = 4 , ToDeviceOp = 5 , FromDeviceOp = 6 , EmptyOp = 7 , OnesOp = 8 , FullOp = 9 , EltwiseOp = 10 , LinearOp = 11 , MatmulOp = 12 , ReductionOp = 13 , EmbeddingOp = 14 , EmbeddingBackwardOp = 15 , SoftmaxOp = 16 , TransposeOp = 17 , Conv2dOp = 18 , ConvTranspose2dOp = 19 , ConcatOp = 20 , ReshapeOp = 21 , SliceOp = 22 , MaxPool2dOp = 23 , DeallocateOp = 24 , AllGatherOp = 25 , ReduceScatterOp = 26 , MeshShardOp = 27 , ArangeOp = 28 , UpdateCacheOp = 29 , FillCacheOp = 30 , PermuteOp = 31 , RepeatOp = 32 , MIN = NONE , MAX = RepeatOp } |
Functions | |
inline ::flatbuffers::Offset< TTNNBinary > | CreateTTNNBinary (::flatbuffers::FlatBufferBuilder &_fbb, const tt::target::Version *version=nullptr, ::flatbuffers::Offset<::flatbuffers::String > ttmlir_git_hash=0, ::flatbuffers::Offset< tt::target::SystemDesc > system_desc=0, ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::ttnn::Program >>> programs=0) |
inline ::flatbuffers::Offset< TTNNBinary > | CreateTTNNBinaryDirect (::flatbuffers::FlatBufferBuilder &_fbb, const tt::target::Version *version=nullptr, const char *ttmlir_git_hash=nullptr, ::flatbuffers::Offset< tt::target::SystemDesc > system_desc=0, const std::vector<::flatbuffers::Offset< tt::target::ttnn::Program >> *programs=nullptr) |
const tt::target::ttnn::TTNNBinary * | GetTTNNBinary (const void *buf) |
const tt::target::ttnn::TTNNBinary * | GetSizePrefixedTTNNBinary (const void *buf) |
const char * | TTNNBinaryIdentifier () |
bool | TTNNBinaryBufferHasIdentifier (const void *buf) |
bool | SizePrefixedTTNNBinaryBufferHasIdentifier (const void *buf) |
bool | VerifyTTNNBinaryBuffer (::flatbuffers::Verifier &verifier) |
bool | VerifySizePrefixedTTNNBinaryBuffer (::flatbuffers::Verifier &verifier) |
const char * | TTNNBinaryExtension () |
void | FinishTTNNBinaryBuffer (::flatbuffers::FlatBufferBuilder &fbb, ::flatbuffers::Offset< tt::target::ttnn::TTNNBinary > root) |
void | FinishSizePrefixedTTNNBinaryBuffer (::flatbuffers::FlatBufferBuilder &fbb, ::flatbuffers::Offset< tt::target::ttnn::TTNNBinary > root) |
const EltwiseOpType(& | EnumValuesEltwiseOpType ())[47] |
const char *const * | EnumNamesEltwiseOpType () |
const char * | EnumNameEltwiseOpType (EltwiseOpType e) |
const EltwiseOpParams(& | EnumValuesEltwiseOpParams ())[3] |
const char *const * | EnumNamesEltwiseOpParams () |
const char * | EnumNameEltwiseOpParams (EltwiseOpParams e) |
bool | VerifyEltwiseOpParams (::flatbuffers::Verifier &verifier, const void *obj, EltwiseOpParams type) |
bool | VerifyEltwiseOpParamsVector (::flatbuffers::Verifier &verifier, const ::flatbuffers::Vector<::flatbuffers::Offset< void >> *values, const ::flatbuffers::Vector< EltwiseOpParams > *types) |
const ReductionOpType(& | EnumValuesReductionOpType ())[3] |
const char *const * | EnumNamesReductionOpType () |
const char * | EnumNameReductionOpType (ReductionOpType e) |
const OpType(& | EnumValuesOpType ())[33] |
const char *const * | EnumNamesOpType () |
const char * | EnumNameOpType (OpType e) |
bool | VerifyOpType (::flatbuffers::Verifier &verifier, const void *obj, OpType type) |
bool | VerifyOpTypeVector (::flatbuffers::Verifier &verifier, const ::flatbuffers::Vector<::flatbuffers::Offset< void >> *values, const ::flatbuffers::Vector< OpType > *types) |
inline ::flatbuffers::Offset< GetDeviceOp > | CreateGetDeviceOp (::flatbuffers::FlatBufferBuilder &_fbb, const tt::target::Dim2d *mesh=nullptr, ::flatbuffers::Offset<::flatbuffers::Vector< uint32_t >> chip_ids=0, ::flatbuffers::Offset< tt::target::DeviceRef > out=0) |
inline ::flatbuffers::Offset< GetDeviceOp > | CreateGetDeviceOpDirect (::flatbuffers::FlatBufferBuilder &_fbb, const tt::target::Dim2d *mesh=nullptr, const std::vector< uint32_t > *chip_ids=nullptr, ::flatbuffers::Offset< tt::target::DeviceRef > out=0) |
inline ::flatbuffers::Offset< ToMemoryConfigOp > | CreateToMemoryConfigOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in0=0, ::flatbuffers::Offset< tt::target::MemoryConfigDesc > memcfg=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0) |
inline ::flatbuffers::Offset< ToLayoutOp > | CreateToLayoutOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, tt::target::TensorLayout layout=tt::target::TensorLayout::RowMajor, ::flatbuffers::Optional< tt::target::DataType > dtype=::flatbuffers::nullopt, ::flatbuffers::Offset< tt::target::MemoryConfigDesc > memcfg=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0) |
inline ::flatbuffers::Offset< TypecastOp > | CreateTypecastOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, tt::target::DataType dtype=tt::target::DataType::Float32, ::flatbuffers::Offset< tt::target::TensorRef > out=0) |
inline ::flatbuffers::Offset< ToDeviceOp > | CreateToDeviceOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, ::flatbuffers::Offset< tt::target::MemoryConfigDesc > memcfg=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0) |
inline ::flatbuffers::Offset< UpdateCacheOp > | CreateUpdateCacheOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > cache=0, ::flatbuffers::Offset< tt::target::TensorRef > input=0, ::flatbuffers::Offset< tt::target::TensorRef > update_index=0, uint32_t batch_offset=0) |
inline ::flatbuffers::Offset< FillCacheOp > | CreateFillCacheOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > cache=0, ::flatbuffers::Offset< tt::target::TensorRef > input=0, uint32_t batch_offset=0) |
inline ::flatbuffers::Offset< FromDeviceOp > | CreateFromDeviceOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0) |
inline ::flatbuffers::Offset< EmptyOp > | CreateEmptyOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> shape=0, tt::target::DataType dtype=tt::target::DataType::Float32, tt::target::TensorLayout layout=tt::target::TensorLayout::RowMajor, uint32_t num_shards=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, ::flatbuffers::Offset< tt::target::MemoryConfigDesc > memcfg=0, ::flatbuffers::Offset< tt::target::DistributionStrategy > strategy=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0) |
inline ::flatbuffers::Offset< EmptyOp > | CreateEmptyOpDirect (::flatbuffers::FlatBufferBuilder &_fbb, const std::vector< int64_t > *shape=nullptr, tt::target::DataType dtype=tt::target::DataType::Float32, tt::target::TensorLayout layout=tt::target::TensorLayout::RowMajor, uint32_t num_shards=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, ::flatbuffers::Offset< tt::target::MemoryConfigDesc > memcfg=0, ::flatbuffers::Offset< tt::target::DistributionStrategy > strategy=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0) |
inline ::flatbuffers::Offset< OnesOp > | CreateOnesOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> shape=0, ::flatbuffers::Optional< tt::target::DataType > dtype=::flatbuffers::nullopt, ::flatbuffers::Optional< tt::target::TensorLayout > layout=::flatbuffers::nullopt, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, ::flatbuffers::Offset< tt::target::MemoryConfigDesc > memcfg=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0) |
inline ::flatbuffers::Offset< OnesOp > | CreateOnesOpDirect (::flatbuffers::FlatBufferBuilder &_fbb, const std::vector< int64_t > *shape=nullptr, ::flatbuffers::Optional< tt::target::DataType > dtype=::flatbuffers::nullopt, ::flatbuffers::Optional< tt::target::TensorLayout > layout=::flatbuffers::nullopt, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, ::flatbuffers::Offset< tt::target::MemoryConfigDesc > memcfg=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0) |
inline ::flatbuffers::Offset< FullOp > | CreateFullOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, float fill_value=0.0f, uint32_t num_shards=0, ::flatbuffers::Offset< tt::target::DistributionStrategy > strategy=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0) |
inline ::flatbuffers::Offset< ArangeOp > | CreateArangeOp (::flatbuffers::FlatBufferBuilder &_fbb, float start=0.0f, float end=0.0f, float step=0.0f, ::flatbuffers::Optional< tt::target::DataType > dtype=::flatbuffers::nullopt, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, ::flatbuffers::Offset< tt::target::MemoryConfigDesc > memcfg=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0) |
inline ::flatbuffers::Offset< ClampOpParams > | CreateClampOpParams (::flatbuffers::FlatBufferBuilder &_fbb, float min=0.0f, float max=0.0f) |
inline ::flatbuffers::Offset< EltwiseOpWithFloatParams > | CreateEltwiseOpWithFloatParams (::flatbuffers::FlatBufferBuilder &_fbb, float parameter=0.0f) |
template<> | |
const tt::target::ttnn::ClampOpParams * | EltwiseOp::params_as< tt::target::ttnn::ClampOpParams > () const |
template<> | |
const tt::target::ttnn::EltwiseOpWithFloatParams * | EltwiseOp::params_as< tt::target::ttnn::EltwiseOpWithFloatParams > () const |
inline ::flatbuffers::Offset< EltwiseOp > | CreateEltwiseOp (::flatbuffers::FlatBufferBuilder &_fbb, tt::target::ttnn::EltwiseOpType type=tt::target::ttnn::EltwiseOpType::Add, ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::TensorRef >>> ins=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, tt::target::ttnn::EltwiseOpParams params_type=tt::target::ttnn::EltwiseOpParams::NONE, ::flatbuffers::Offset< void > params=0) |
inline ::flatbuffers::Offset< EltwiseOp > | CreateEltwiseOpDirect (::flatbuffers::FlatBufferBuilder &_fbb, tt::target::ttnn::EltwiseOpType type=tt::target::ttnn::EltwiseOpType::Add, const std::vector<::flatbuffers::Offset< tt::target::TensorRef >> *ins=nullptr, ::flatbuffers::Offset< tt::target::TensorRef > out=0, tt::target::ttnn::EltwiseOpParams params_type=tt::target::ttnn::EltwiseOpParams::NONE, ::flatbuffers::Offset< void > params=0) |
inline ::flatbuffers::Offset< ReductionOp > | CreateReductionOp (::flatbuffers::FlatBufferBuilder &_fbb, tt::target::ttnn::ReductionOpType type=tt::target::ttnn::ReductionOpType::Sum, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> dim_arg=0, bool keep_dim=false) |
inline ::flatbuffers::Offset< ReductionOp > | CreateReductionOpDirect (::flatbuffers::FlatBufferBuilder &_fbb, tt::target::ttnn::ReductionOpType type=tt::target::ttnn::ReductionOpType::Sum, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, const std::vector< int32_t > *dim_arg=nullptr, bool keep_dim=false) |
inline ::flatbuffers::Offset< EmbeddingOp > | CreateEmbeddingOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > input=0, ::flatbuffers::Offset< tt::target::TensorRef > weight=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0) |
inline ::flatbuffers::Offset< EmbeddingBackwardOp > | CreateEmbeddingBackwardOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > input=0, ::flatbuffers::Offset< tt::target::TensorRef > weight=0, ::flatbuffers::Offset< tt::target::TensorRef > in_grad=0, ::flatbuffers::Optional< tt::target::DataType > dtype=::flatbuffers::nullopt, ::flatbuffers::Offset< tt::target::MemoryConfigDesc > memcfg=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0) |
inline ::flatbuffers::Offset< SoftmaxOp > | CreateSoftmaxOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, int32_t dimension=0) |
inline ::flatbuffers::Offset< TransposeOp > | CreateTransposeOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, int32_t dim0=0, int32_t dim1=0) |
inline ::flatbuffers::Offset< ConcatOp > | CreateConcatOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::TensorRef >>> inputs=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, int32_t dim=0) |
inline ::flatbuffers::Offset< ConcatOp > | CreateConcatOpDirect (::flatbuffers::FlatBufferBuilder &_fbb, const std::vector<::flatbuffers::Offset< tt::target::TensorRef >> *inputs=nullptr, ::flatbuffers::Offset< tt::target::TensorRef > out=0, int32_t dim=0) |
inline ::flatbuffers::Offset< ReshapeOp > | CreateReshapeOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> shape=0) |
inline ::flatbuffers::Offset< ReshapeOp > | CreateReshapeOpDirect (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, const std::vector< int32_t > *shape=nullptr) |
inline ::flatbuffers::Offset< RepeatOp > | CreateRepeatOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, ::flatbuffers::Offset<::flatbuffers::Vector< uint32_t >> shape=0) |
inline ::flatbuffers::Offset< RepeatOp > | CreateRepeatOpDirect (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, const std::vector< uint32_t > *shape=nullptr) |
inline ::flatbuffers::Offset< SliceOp > | CreateSliceOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, ::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> begins=0, ::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> ends=0, ::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> step=0) |
inline ::flatbuffers::Offset< SliceOp > | CreateSliceOpDirect (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, const std::vector< int64_t > *begins=nullptr, const std::vector< int64_t > *ends=nullptr, const std::vector< int64_t > *step=nullptr) |
inline ::flatbuffers::Offset< LinearOp > | CreateLinearOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in0=0, ::flatbuffers::Offset< tt::target::TensorRef > in1=0, ::flatbuffers::Offset< tt::target::TensorRef > bias=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0) |
inline ::flatbuffers::Offset< MatmulOp > | CreateMatmulOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in0=0, ::flatbuffers::Offset< tt::target::TensorRef > in1=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0) |
inline ::flatbuffers::Offset< Conv2dOp > | CreateConv2dOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > input=0, ::flatbuffers::Offset< tt::target::TensorRef > weight=0, ::flatbuffers::Offset< tt::target::TensorRef > bias=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, uint32_t in_channels=0, uint32_t out_channels=0, uint32_t batch_size=0, uint32_t input_height=0, uint32_t input_width=0, uint32_t kernel_height=0, uint32_t kernel_width=0, uint32_t stride_height=0, uint32_t stride_width=0, uint32_t padding_height=0, uint32_t padding_width=0, uint32_t dilation_height=0, uint32_t dilation_width=0, uint32_t groups=0) |
inline ::flatbuffers::Offset< ConvTranspose2dOp > | CreateConvTranspose2dOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > input=0, ::flatbuffers::Offset< tt::target::TensorRef > weight=0, ::flatbuffers::Offset< tt::target::TensorRef > bias=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, uint32_t in_channels=0, uint32_t out_channels=0, uint32_t batch_size=0, uint32_t input_height=0, uint32_t input_width=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> kernel_size=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> stride=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> padding=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> output_padding=0, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> dilation=0, uint32_t groups=0) |
inline ::flatbuffers::Offset< ConvTranspose2dOp > | CreateConvTranspose2dOpDirect (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > input=0, ::flatbuffers::Offset< tt::target::TensorRef > weight=0, ::flatbuffers::Offset< tt::target::TensorRef > bias=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, uint32_t in_channels=0, uint32_t out_channels=0, uint32_t batch_size=0, uint32_t input_height=0, uint32_t input_width=0, const std::vector< int32_t > *kernel_size=nullptr, const std::vector< int32_t > *stride=nullptr, const std::vector< int32_t > *padding=nullptr, const std::vector< int32_t > *output_padding=nullptr, const std::vector< int32_t > *dilation=nullptr, uint32_t groups=0) |
inline ::flatbuffers::Offset< MaxPool2dOp > | CreateMaxPool2dOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, uint32_t batch_size=0, uint32_t input_height=0, uint32_t input_width=0, uint32_t channels=0, uint32_t kernel_height=0, uint32_t kernel_width=0, uint32_t stride_height=0, uint32_t stride_width=0, uint32_t dilation_height=0, uint32_t dilation_width=0, bool ceil_mode=false, uint32_t padding_height=0, uint32_t padding_width=0) |
inline ::flatbuffers::Offset< DeallocateOp > | CreateDeallocateOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, bool force=false) |
inline ::flatbuffers::Offset< AllGatherOp > | CreateAllGatherOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, uint32_t dim=0, uint32_t num_links=0) |
inline ::flatbuffers::Offset< PermuteOp > | CreatePermuteOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> permutation=0, ::flatbuffers::Offset< tt::target::MemoryConfigDesc > memory_config=0, float pad_value=0.0f, ::flatbuffers::Offset< tt::target::TensorRef > out=0) |
inline ::flatbuffers::Offset< PermuteOp > | CreatePermuteOpDirect (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, const std::vector< int64_t > *permutation=nullptr, ::flatbuffers::Offset< tt::target::MemoryConfigDesc > memory_config=0, float pad_value=0.0f, ::flatbuffers::Offset< tt::target::TensorRef > out=0) |
inline ::flatbuffers::Offset< ReduceScatterOp > | CreateReduceScatterOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, uint32_t scatter_split_dim=0, uint32_t math_op=0, uint32_t num_links=0) |
inline ::flatbuffers::Offset< MeshShardOp > | CreateMeshShardOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, tt::target::MeshShardDirection shard_direction=tt::target::MeshShardDirection::FullToShardShape, tt::target::MeshShardType shard_type=tt::target::MeshShardType::Manual, ::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> shard_shape=0) |
inline ::flatbuffers::Offset< MeshShardOp > | CreateMeshShardOpDirect (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::TensorRef > in=0, ::flatbuffers::Offset< tt::target::TensorRef > out=0, ::flatbuffers::Offset< tt::target::DeviceRef > device=0, tt::target::MeshShardDirection shard_direction=tt::target::MeshShardDirection::FullToShardShape, tt::target::MeshShardType shard_type=tt::target::MeshShardType::Manual, const std::vector< int64_t > *shard_shape=nullptr) |
template<> | |
const tt::target::ttnn::GetDeviceOp * | Operation::type_as< tt::target::ttnn::GetDeviceOp > () const |
template<> | |
const tt::target::ttnn::ToMemoryConfigOp * | Operation::type_as< tt::target::ttnn::ToMemoryConfigOp > () const |
template<> | |
const tt::target::ttnn::ToLayoutOp * | Operation::type_as< tt::target::ttnn::ToLayoutOp > () const |
template<> | |
const tt::target::ttnn::TypecastOp * | Operation::type_as< tt::target::ttnn::TypecastOp > () const |
template<> | |
const tt::target::ttnn::ToDeviceOp * | Operation::type_as< tt::target::ttnn::ToDeviceOp > () const |
template<> | |
const tt::target::ttnn::FromDeviceOp * | Operation::type_as< tt::target::ttnn::FromDeviceOp > () const |
template<> | |
const tt::target::ttnn::EmptyOp * | Operation::type_as< tt::target::ttnn::EmptyOp > () const |
template<> | |
const tt::target::ttnn::OnesOp * | Operation::type_as< tt::target::ttnn::OnesOp > () const |
template<> | |
const tt::target::ttnn::FullOp * | Operation::type_as< tt::target::ttnn::FullOp > () const |
template<> | |
const tt::target::ttnn::EltwiseOp * | Operation::type_as< tt::target::ttnn::EltwiseOp > () const |
template<> | |
const tt::target::ttnn::LinearOp * | Operation::type_as< tt::target::ttnn::LinearOp > () const |
template<> | |
const tt::target::ttnn::MatmulOp * | Operation::type_as< tt::target::ttnn::MatmulOp > () const |
template<> | |
const tt::target::ttnn::ReductionOp * | Operation::type_as< tt::target::ttnn::ReductionOp > () const |
template<> | |
const tt::target::ttnn::EmbeddingOp * | Operation::type_as< tt::target::ttnn::EmbeddingOp > () const |
template<> | |
const tt::target::ttnn::EmbeddingBackwardOp * | Operation::type_as< tt::target::ttnn::EmbeddingBackwardOp > () const |
template<> | |
const tt::target::ttnn::SoftmaxOp * | Operation::type_as< tt::target::ttnn::SoftmaxOp > () const |
template<> | |
const tt::target::ttnn::TransposeOp * | Operation::type_as< tt::target::ttnn::TransposeOp > () const |
template<> | |
const tt::target::ttnn::Conv2dOp * | Operation::type_as< tt::target::ttnn::Conv2dOp > () const |
template<> | |
const tt::target::ttnn::ConvTranspose2dOp * | Operation::type_as< tt::target::ttnn::ConvTranspose2dOp > () const |
template<> | |
const tt::target::ttnn::ConcatOp * | Operation::type_as< tt::target::ttnn::ConcatOp > () const |
template<> | |
const tt::target::ttnn::ReshapeOp * | Operation::type_as< tt::target::ttnn::ReshapeOp > () const |
template<> | |
const tt::target::ttnn::SliceOp * | Operation::type_as< tt::target::ttnn::SliceOp > () const |
template<> | |
const tt::target::ttnn::MaxPool2dOp * | Operation::type_as< tt::target::ttnn::MaxPool2dOp > () const |
template<> | |
const tt::target::ttnn::DeallocateOp * | Operation::type_as< tt::target::ttnn::DeallocateOp > () const |
template<> | |
const tt::target::ttnn::AllGatherOp * | Operation::type_as< tt::target::ttnn::AllGatherOp > () const |
template<> | |
const tt::target::ttnn::ReduceScatterOp * | Operation::type_as< tt::target::ttnn::ReduceScatterOp > () const |
template<> | |
const tt::target::ttnn::MeshShardOp * | Operation::type_as< tt::target::ttnn::MeshShardOp > () const |
template<> | |
const tt::target::ttnn::ArangeOp * | Operation::type_as< tt::target::ttnn::ArangeOp > () const |
template<> | |
const tt::target::ttnn::UpdateCacheOp * | Operation::type_as< tt::target::ttnn::UpdateCacheOp > () const |
template<> | |
const tt::target::ttnn::FillCacheOp * | Operation::type_as< tt::target::ttnn::FillCacheOp > () const |
template<> | |
const tt::target::ttnn::PermuteOp * | Operation::type_as< tt::target::ttnn::PermuteOp > () const |
template<> | |
const tt::target::ttnn::RepeatOp * | Operation::type_as< tt::target::ttnn::RepeatOp > () const |
inline ::flatbuffers::Offset< Operation > | CreateOperation (::flatbuffers::FlatBufferBuilder &_fbb, tt::target::ttnn::OpType type_type=tt::target::ttnn::OpType::NONE, ::flatbuffers::Offset< void > type=0, ::flatbuffers::Offset<::flatbuffers::String > debug_info=0, ::flatbuffers::Offset<::flatbuffers::String > loc_info=0) |
inline ::flatbuffers::Offset< Operation > | CreateOperationDirect (::flatbuffers::FlatBufferBuilder &_fbb, tt::target::ttnn::OpType type_type=tt::target::ttnn::OpType::NONE, ::flatbuffers::Offset< void > type=0, const char *debug_info=nullptr, const char *loc_info=nullptr) |
inline ::flatbuffers::Offset< Program > | CreateProgram (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset<::flatbuffers::String > name=0, ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::TensorRef >>> inputs=0, ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::TensorRef >>> outputs=0, ::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::ttnn::Operation >>> operations=0, ::flatbuffers::Offset< tt::target::DebugInfo > debug_info=0) |
inline ::flatbuffers::Offset< Program > | CreateProgramDirect (::flatbuffers::FlatBufferBuilder &_fbb, const char *name=nullptr, const std::vector<::flatbuffers::Offset< tt::target::TensorRef >> *inputs=nullptr, const std::vector<::flatbuffers::Offset< tt::target::TensorRef >> *outputs=nullptr, const std::vector<::flatbuffers::Offset< tt::target::ttnn::Operation >> *operations=nullptr, ::flatbuffers::Offset< tt::target::DebugInfo > debug_info=0) |
|
strong |
|
strong |
|
strong |
|
strong |
inline ::flatbuffers::Offset<AllGatherOp> tt::target::ttnn::CreateAllGatherOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::TensorRef > | in = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 , |
||
::flatbuffers::Offset< tt::target::DeviceRef > | device = 0 , |
||
uint32_t | dim = 0 , |
||
uint32_t | num_links = 0 |
||
) |
inline ::flatbuffers::Offset<ArangeOp> tt::target::ttnn::CreateArangeOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
float | start = 0.0f , |
||
float | end = 0.0f , |
||
float | step = 0.0f , |
||
::flatbuffers::Optional< tt::target::DataType > | dtype = ::flatbuffers::nullopt , |
||
::flatbuffers::Offset< tt::target::DeviceRef > | device = 0 , |
||
::flatbuffers::Offset< tt::target::MemoryConfigDesc > | memcfg = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 |
||
) |
inline ::flatbuffers::Offset<ClampOpParams> tt::target::ttnn::CreateClampOpParams | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
float | min = 0.0f , |
||
float | max = 0.0f |
||
) |
inline ::flatbuffers::Offset<ConcatOp> tt::target::ttnn::CreateConcatOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::TensorRef >>> | inputs = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 , |
||
int32_t | dim = 0 |
||
) |
inline ::flatbuffers::Offset<ConcatOp> tt::target::ttnn::CreateConcatOpDirect | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
const std::vector<::flatbuffers::Offset< tt::target::TensorRef >> * | inputs = nullptr , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 , |
||
int32_t | dim = 0 |
||
) |
inline ::flatbuffers::Offset<Conv2dOp> tt::target::ttnn::CreateConv2dOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::TensorRef > | input = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | weight = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | bias = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 , |
||
::flatbuffers::Offset< tt::target::DeviceRef > | device = 0 , |
||
uint32_t | in_channels = 0 , |
||
uint32_t | out_channels = 0 , |
||
uint32_t | batch_size = 0 , |
||
uint32_t | input_height = 0 , |
||
uint32_t | input_width = 0 , |
||
uint32_t | kernel_height = 0 , |
||
uint32_t | kernel_width = 0 , |
||
uint32_t | stride_height = 0 , |
||
uint32_t | stride_width = 0 , |
||
uint32_t | padding_height = 0 , |
||
uint32_t | padding_width = 0 , |
||
uint32_t | dilation_height = 0 , |
||
uint32_t | dilation_width = 0 , |
||
uint32_t | groups = 0 |
||
) |
inline ::flatbuffers::Offset<ConvTranspose2dOp> tt::target::ttnn::CreateConvTranspose2dOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::TensorRef > | input = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | weight = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | bias = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 , |
||
::flatbuffers::Offset< tt::target::DeviceRef > | device = 0 , |
||
uint32_t | in_channels = 0 , |
||
uint32_t | out_channels = 0 , |
||
uint32_t | batch_size = 0 , |
||
uint32_t | input_height = 0 , |
||
uint32_t | input_width = 0 , |
||
::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> | kernel_size = 0 , |
||
::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> | stride = 0 , |
||
::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> | padding = 0 , |
||
::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> | output_padding = 0 , |
||
::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> | dilation = 0 , |
||
uint32_t | groups = 0 |
||
) |
inline ::flatbuffers::Offset<ConvTranspose2dOp> tt::target::ttnn::CreateConvTranspose2dOpDirect | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::TensorRef > | input = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | weight = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | bias = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 , |
||
::flatbuffers::Offset< tt::target::DeviceRef > | device = 0 , |
||
uint32_t | in_channels = 0 , |
||
uint32_t | out_channels = 0 , |
||
uint32_t | batch_size = 0 , |
||
uint32_t | input_height = 0 , |
||
uint32_t | input_width = 0 , |
||
const std::vector< int32_t > * | kernel_size = nullptr , |
||
const std::vector< int32_t > * | stride = nullptr , |
||
const std::vector< int32_t > * | padding = nullptr , |
||
const std::vector< int32_t > * | output_padding = nullptr , |
||
const std::vector< int32_t > * | dilation = nullptr , |
||
uint32_t | groups = 0 |
||
) |
inline ::flatbuffers::Offset<DeallocateOp> tt::target::ttnn::CreateDeallocateOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::TensorRef > | in = 0 , |
||
bool | force = false |
||
) |
inline ::flatbuffers::Offset<EltwiseOp> tt::target::ttnn::CreateEltwiseOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
tt::target::ttnn::EltwiseOpType | type = tt::target::ttnn::EltwiseOpType::Add , |
||
::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::TensorRef >>> | ins = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 , |
||
tt::target::ttnn::EltwiseOpParams | params_type = tt::target::ttnn::EltwiseOpParams::NONE , |
||
::flatbuffers::Offset< void > | params = 0 |
||
) |
inline ::flatbuffers::Offset<EltwiseOp> tt::target::ttnn::CreateEltwiseOpDirect | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
tt::target::ttnn::EltwiseOpType | type = tt::target::ttnn::EltwiseOpType::Add , |
||
const std::vector<::flatbuffers::Offset< tt::target::TensorRef >> * | ins = nullptr , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 , |
||
tt::target::ttnn::EltwiseOpParams | params_type = tt::target::ttnn::EltwiseOpParams::NONE , |
||
::flatbuffers::Offset< void > | params = 0 |
||
) |
inline ::flatbuffers::Offset<EltwiseOpWithFloatParams> tt::target::ttnn::CreateEltwiseOpWithFloatParams | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
float | parameter = 0.0f |
||
) |
inline ::flatbuffers::Offset<EmbeddingBackwardOp> tt::target::ttnn::CreateEmbeddingBackwardOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::TensorRef > | input = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | weight = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | in_grad = 0 , |
||
::flatbuffers::Optional< tt::target::DataType > | dtype = ::flatbuffers::nullopt , |
||
::flatbuffers::Offset< tt::target::MemoryConfigDesc > | memcfg = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 |
||
) |
inline ::flatbuffers::Offset<EmbeddingOp> tt::target::ttnn::CreateEmbeddingOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::TensorRef > | input = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | weight = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 |
||
) |
inline ::flatbuffers::Offset<EmptyOp> tt::target::ttnn::CreateEmptyOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> | shape = 0 , |
||
tt::target::DataType | dtype = tt::target::DataType::Float32 , |
||
tt::target::TensorLayout | layout = tt::target::TensorLayout::RowMajor , |
||
uint32_t | num_shards = 0 , |
||
::flatbuffers::Offset< tt::target::DeviceRef > | device = 0 , |
||
::flatbuffers::Offset< tt::target::MemoryConfigDesc > | memcfg = 0 , |
||
::flatbuffers::Offset< tt::target::DistributionStrategy > | strategy = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 |
||
) |
inline ::flatbuffers::Offset<EmptyOp> tt::target::ttnn::CreateEmptyOpDirect | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
const std::vector< int64_t > * | shape = nullptr , |
||
tt::target::DataType | dtype = tt::target::DataType::Float32 , |
||
tt::target::TensorLayout | layout = tt::target::TensorLayout::RowMajor , |
||
uint32_t | num_shards = 0 , |
||
::flatbuffers::Offset< tt::target::DeviceRef > | device = 0 , |
||
::flatbuffers::Offset< tt::target::MemoryConfigDesc > | memcfg = 0 , |
||
::flatbuffers::Offset< tt::target::DistributionStrategy > | strategy = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 |
||
) |
inline ::flatbuffers::Offset<FillCacheOp> tt::target::ttnn::CreateFillCacheOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::TensorRef > | cache = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | input = 0 , |
||
uint32_t | batch_offset = 0 |
||
) |
inline ::flatbuffers::Offset<FromDeviceOp> tt::target::ttnn::CreateFromDeviceOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::TensorRef > | in = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 |
||
) |
inline ::flatbuffers::Offset<FullOp> tt::target::ttnn::CreateFullOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::DeviceRef > | device = 0 , |
||
float | fill_value = 0.0f , |
||
uint32_t | num_shards = 0 , |
||
::flatbuffers::Offset< tt::target::DistributionStrategy > | strategy = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 |
||
) |
inline ::flatbuffers::Offset<GetDeviceOp> tt::target::ttnn::CreateGetDeviceOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
const tt::target::Dim2d * | mesh = nullptr , |
||
::flatbuffers::Offset<::flatbuffers::Vector< uint32_t >> | chip_ids = 0 , |
||
::flatbuffers::Offset< tt::target::DeviceRef > | out = 0 |
||
) |
inline ::flatbuffers::Offset<GetDeviceOp> tt::target::ttnn::CreateGetDeviceOpDirect | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
const tt::target::Dim2d * | mesh = nullptr , |
||
const std::vector< uint32_t > * | chip_ids = nullptr , |
||
::flatbuffers::Offset< tt::target::DeviceRef > | out = 0 |
||
) |
inline ::flatbuffers::Offset<LinearOp> tt::target::ttnn::CreateLinearOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::TensorRef > | in0 = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | in1 = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | bias = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 |
||
) |
inline ::flatbuffers::Offset<MatmulOp> tt::target::ttnn::CreateMatmulOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::TensorRef > | in0 = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | in1 = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 |
||
) |
inline ::flatbuffers::Offset<MaxPool2dOp> tt::target::ttnn::CreateMaxPool2dOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::TensorRef > | in = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 , |
||
::flatbuffers::Offset< tt::target::DeviceRef > | device = 0 , |
||
uint32_t | batch_size = 0 , |
||
uint32_t | input_height = 0 , |
||
uint32_t | input_width = 0 , |
||
uint32_t | channels = 0 , |
||
uint32_t | kernel_height = 0 , |
||
uint32_t | kernel_width = 0 , |
||
uint32_t | stride_height = 0 , |
||
uint32_t | stride_width = 0 , |
||
uint32_t | dilation_height = 0 , |
||
uint32_t | dilation_width = 0 , |
||
bool | ceil_mode = false , |
||
uint32_t | padding_height = 0 , |
||
uint32_t | padding_width = 0 |
||
) |
inline ::flatbuffers::Offset<MeshShardOp> tt::target::ttnn::CreateMeshShardOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::TensorRef > | in = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 , |
||
::flatbuffers::Offset< tt::target::DeviceRef > | device = 0 , |
||
tt::target::MeshShardDirection | shard_direction = tt::target::MeshShardDirection::FullToShardShape , |
||
tt::target::MeshShardType | shard_type = tt::target::MeshShardType::Manual , |
||
::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> | shard_shape = 0 |
||
) |
inline ::flatbuffers::Offset<MeshShardOp> tt::target::ttnn::CreateMeshShardOpDirect | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::TensorRef > | in = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 , |
||
::flatbuffers::Offset< tt::target::DeviceRef > | device = 0 , |
||
tt::target::MeshShardDirection | shard_direction = tt::target::MeshShardDirection::FullToShardShape , |
||
tt::target::MeshShardType | shard_type = tt::target::MeshShardType::Manual , |
||
const std::vector< int64_t > * | shard_shape = nullptr |
||
) |
inline ::flatbuffers::Offset<OnesOp> tt::target::ttnn::CreateOnesOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> | shape = 0 , |
||
::flatbuffers::Optional< tt::target::DataType > | dtype = ::flatbuffers::nullopt , |
||
::flatbuffers::Optional< tt::target::TensorLayout > | layout = ::flatbuffers::nullopt , |
||
::flatbuffers::Offset< tt::target::DeviceRef > | device = 0 , |
||
::flatbuffers::Offset< tt::target::MemoryConfigDesc > | memcfg = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 |
||
) |
inline ::flatbuffers::Offset<OnesOp> tt::target::ttnn::CreateOnesOpDirect | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
const std::vector< int64_t > * | shape = nullptr , |
||
::flatbuffers::Optional< tt::target::DataType > | dtype = ::flatbuffers::nullopt , |
||
::flatbuffers::Optional< tt::target::TensorLayout > | layout = ::flatbuffers::nullopt , |
||
::flatbuffers::Offset< tt::target::DeviceRef > | device = 0 , |
||
::flatbuffers::Offset< tt::target::MemoryConfigDesc > | memcfg = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 |
||
) |
inline ::flatbuffers::Offset<Operation> tt::target::ttnn::CreateOperation | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
tt::target::ttnn::OpType | type_type = tt::target::ttnn::OpType::NONE , |
||
::flatbuffers::Offset< void > | type = 0 , |
||
::flatbuffers::Offset<::flatbuffers::String > | debug_info = 0 , |
||
::flatbuffers::Offset<::flatbuffers::String > | loc_info = 0 |
||
) |
inline ::flatbuffers::Offset<Operation> tt::target::ttnn::CreateOperationDirect | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
tt::target::ttnn::OpType | type_type = tt::target::ttnn::OpType::NONE , |
||
::flatbuffers::Offset< void > | type = 0 , |
||
const char * | debug_info = nullptr , |
||
const char * | loc_info = nullptr |
||
) |
inline ::flatbuffers::Offset<PermuteOp> tt::target::ttnn::CreatePermuteOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::TensorRef > | in = 0 , |
||
::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> | permutation = 0 , |
||
::flatbuffers::Offset< tt::target::MemoryConfigDesc > | memory_config = 0 , |
||
float | pad_value = 0.0f , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 |
||
) |
inline ::flatbuffers::Offset<PermuteOp> tt::target::ttnn::CreatePermuteOpDirect | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::TensorRef > | in = 0 , |
||
const std::vector< int64_t > * | permutation = nullptr , |
||
::flatbuffers::Offset< tt::target::MemoryConfigDesc > | memory_config = 0 , |
||
float | pad_value = 0.0f , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 |
||
) |
inline ::flatbuffers::Offset<Program> tt::target::ttnn::CreateProgram | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset<::flatbuffers::String > | name = 0 , |
||
::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::TensorRef >>> | inputs = 0 , |
||
::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::TensorRef >>> | outputs = 0 , |
||
::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::ttnn::Operation >>> | operations = 0 , |
||
::flatbuffers::Offset< tt::target::DebugInfo > | debug_info = 0 |
||
) |
inline ::flatbuffers::Offset<Program> tt::target::ttnn::CreateProgramDirect | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
const char * | name = nullptr , |
||
const std::vector<::flatbuffers::Offset< tt::target::TensorRef >> * | inputs = nullptr , |
||
const std::vector<::flatbuffers::Offset< tt::target::TensorRef >> * | outputs = nullptr , |
||
const std::vector<::flatbuffers::Offset< tt::target::ttnn::Operation >> * | operations = nullptr , |
||
::flatbuffers::Offset< tt::target::DebugInfo > | debug_info = 0 |
||
) |
inline ::flatbuffers::Offset<ReduceScatterOp> tt::target::ttnn::CreateReduceScatterOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::TensorRef > | in = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 , |
||
::flatbuffers::Offset< tt::target::DeviceRef > | device = 0 , |
||
uint32_t | scatter_split_dim = 0 , |
||
uint32_t | math_op = 0 , |
||
uint32_t | num_links = 0 |
||
) |
inline ::flatbuffers::Offset<ReductionOp> tt::target::ttnn::CreateReductionOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
tt::target::ttnn::ReductionOpType | type = tt::target::ttnn::ReductionOpType::Sum , |
||
::flatbuffers::Offset< tt::target::TensorRef > | in = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 , |
||
::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> | dim_arg = 0 , |
||
bool | keep_dim = false |
||
) |
inline ::flatbuffers::Offset<ReductionOp> tt::target::ttnn::CreateReductionOpDirect | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
tt::target::ttnn::ReductionOpType | type = tt::target::ttnn::ReductionOpType::Sum , |
||
::flatbuffers::Offset< tt::target::TensorRef > | in = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 , |
||
const std::vector< int32_t > * | dim_arg = nullptr , |
||
bool | keep_dim = false |
||
) |
inline ::flatbuffers::Offset<RepeatOp> tt::target::ttnn::CreateRepeatOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::TensorRef > | in = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 , |
||
::flatbuffers::Offset<::flatbuffers::Vector< uint32_t >> | shape = 0 |
||
) |
inline ::flatbuffers::Offset<RepeatOp> tt::target::ttnn::CreateRepeatOpDirect | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::TensorRef > | in = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 , |
||
const std::vector< uint32_t > * | shape = nullptr |
||
) |
inline ::flatbuffers::Offset<ReshapeOp> tt::target::ttnn::CreateReshapeOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::TensorRef > | in = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 , |
||
::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> | shape = 0 |
||
) |
inline ::flatbuffers::Offset<ReshapeOp> tt::target::ttnn::CreateReshapeOpDirect | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::TensorRef > | in = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 , |
||
const std::vector< int32_t > * | shape = nullptr |
||
) |
inline ::flatbuffers::Offset<SliceOp> tt::target::ttnn::CreateSliceOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::TensorRef > | in = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 , |
||
::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> | begins = 0 , |
||
::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> | ends = 0 , |
||
::flatbuffers::Offset<::flatbuffers::Vector< int64_t >> | step = 0 |
||
) |
inline ::flatbuffers::Offset<SliceOp> tt::target::ttnn::CreateSliceOpDirect | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::TensorRef > | in = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 , |
||
const std::vector< int64_t > * | begins = nullptr , |
||
const std::vector< int64_t > * | ends = nullptr , |
||
const std::vector< int64_t > * | step = nullptr |
||
) |
inline ::flatbuffers::Offset<SoftmaxOp> tt::target::ttnn::CreateSoftmaxOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::TensorRef > | in = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 , |
||
int32_t | dimension = 0 |
||
) |
inline ::flatbuffers::Offset<ToDeviceOp> tt::target::ttnn::CreateToDeviceOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::TensorRef > | in = 0 , |
||
::flatbuffers::Offset< tt::target::DeviceRef > | device = 0 , |
||
::flatbuffers::Offset< tt::target::MemoryConfigDesc > | memcfg = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 |
||
) |
inline ::flatbuffers::Offset<ToLayoutOp> tt::target::ttnn::CreateToLayoutOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::TensorRef > | in = 0 , |
||
tt::target::TensorLayout | layout = tt::target::TensorLayout::RowMajor , |
||
::flatbuffers::Optional< tt::target::DataType > | dtype = ::flatbuffers::nullopt , |
||
::flatbuffers::Offset< tt::target::MemoryConfigDesc > | memcfg = 0 , |
||
::flatbuffers::Offset< tt::target::DeviceRef > | device = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 |
||
) |
inline ::flatbuffers::Offset<ToMemoryConfigOp> tt::target::ttnn::CreateToMemoryConfigOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::TensorRef > | in0 = 0 , |
||
::flatbuffers::Offset< tt::target::MemoryConfigDesc > | memcfg = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 |
||
) |
inline ::flatbuffers::Offset<TransposeOp> tt::target::ttnn::CreateTransposeOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::TensorRef > | in = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 , |
||
int32_t | dim0 = 0 , |
||
int32_t | dim1 = 0 |
||
) |
inline ::flatbuffers::Offset<TTNNBinary> tt::target::ttnn::CreateTTNNBinary | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
const tt::target::Version * | version = nullptr , |
||
::flatbuffers::Offset<::flatbuffers::String > | ttmlir_git_hash = 0 , |
||
::flatbuffers::Offset< tt::target::SystemDesc > | system_desc = 0 , |
||
::flatbuffers::Offset<::flatbuffers::Vector<::flatbuffers::Offset< tt::target::ttnn::Program >>> | programs = 0 |
||
) |
inline ::flatbuffers::Offset<TTNNBinary> tt::target::ttnn::CreateTTNNBinaryDirect | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
const tt::target::Version * | version = nullptr , |
||
const char * | ttmlir_git_hash = nullptr , |
||
::flatbuffers::Offset< tt::target::SystemDesc > | system_desc = 0 , |
||
const std::vector<::flatbuffers::Offset< tt::target::ttnn::Program >> * | programs = nullptr |
||
) |
inline ::flatbuffers::Offset<TypecastOp> tt::target::ttnn::CreateTypecastOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::TensorRef > | in = 0 , |
||
tt::target::DataType | dtype = tt::target::DataType::Float32 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | out = 0 |
||
) |
inline ::flatbuffers::Offset<UpdateCacheOp> tt::target::ttnn::CreateUpdateCacheOp | ( | ::flatbuffers::FlatBufferBuilder & | _fbb, |
::flatbuffers::Offset< tt::target::TensorRef > | cache = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | input = 0 , |
||
::flatbuffers::Offset< tt::target::TensorRef > | update_index = 0 , |
||
uint32_t | batch_offset = 0 |
||
) |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |