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 , LogicalNot = 22 , Cbrt = 23 , Minimum = 24 , Ceil = 25 , Sin = 26 , Cos = 27 , Log = 28 , Log1p = 29 , Expm1 = 30 , Sign = 31 , Remainder = 32 , IsFinite = 33 , Floor = 34 , Where = 35 , Gelu = 36 , LogicalXor = 37 , Clamp = 38 , LeakyRelu = 39 , MIN = Add , MAX = LeakyRelu } |
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 , FullOp = 8 , EltwiseOp = 9 , MatmulOp = 10 , ReductionOp = 11 , EmbeddingOp = 12 , SoftmaxOp = 13 , TransposeOp = 14 , Conv2dOp = 15 , ConcatOp = 16 , ReshapeOp = 17 , SliceOp = 18 , MaxPool2dOp = 19 , DeallocateOp = 20 , AllGatherOp = 21 , MIN = NONE , MAX = AllGatherOp } |
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 ())[40] |
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 ())[22] |
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< 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< 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< 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< 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< 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< 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< 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, uint32_t dim=0, uint32_t num_links=0) |
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::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::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::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::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 |
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) |
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) |
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 , |
||
uint32_t | dim = 0 , |
||
uint32_t | num_links = 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<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<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<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<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<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 |
||
) |
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 |
||
) |
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<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<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 |
|
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 |