TT-MLIR
Classes | Namespaces | Enumerations | Functions | Variables
reduction_generated.h File Reference
#include "flatbuffers/flatbuffers.h"
#include "ttmlir/Target/TTNN/types_generated.h"

Go to the source code of this file.

Classes

struct  tt::target::ttnn::FLATBUFFERS_FINAL_CLASS
 
struct  tt::target::ttnn::ReductionOpBuilder
 
struct  tt::target::ttnn::ReductionOp::Traits
 
struct  tt::target::ttnn::ReductionArgMaxOpBuilder
 
struct  tt::target::ttnn::ReductionArgMaxOp::Traits
 
struct  tt::target::ttnn::ReductionProdOpBuilder
 
struct  tt::target::ttnn::ReductionProdOp::Traits
 

Namespaces

 tt
 
 tt::target
 
 tt::target::ttnn
 

Enumerations

enum class  tt::target::ttnn::ReductionOpType : uint32_t {
  tt::target::ttnn::Sum = 0 , tt::target::ttnn::Mean = 1 , tt::target::ttnn::Max = 2 , tt::target::ttnn::Min = 3 ,
  tt::target::ttnn::MIN = Sum , tt::target::ttnn::MAX = Min
}
 

Functions

const ReductionOpType(& tt::target::ttnn::EnumValuesReductionOpType ())[4]
 
const char *const * tt::target::ttnn::EnumNamesReductionOpType ()
 
const char * tt::target::ttnn::EnumNameReductionOpType (ReductionOpType e)
 
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::ttnn::TensorRef > in=0, ::flatbuffers::Offset< tt::target::ttnn::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::ttnn::TensorRef > in=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > out=0, const std::vector< int32_t > *dim_arg=nullptr, bool keep_dim=false)
 
inline ::flatbuffers::Offset< ReductionArgMaxOp > tt::target::ttnn::CreateReductionArgMaxOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > in=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > out=0, ::flatbuffers::Optional< int32_t > dim=::flatbuffers::nullopt, bool keep_dim=false, bool use_multicore=false, ::flatbuffers::Offset< tt::target::ttnn::MemoryConfig > memcfg=0)
 
inline ::flatbuffers::Offset< ReductionProdOp > tt::target::ttnn::CreateReductionProdOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > in=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > out=0, bool all_dimensions=false, int64_t dim_arg=0, bool keep_dim=false, ::flatbuffers::Offset< tt::target::ttnn::MemoryConfig > memcfg=0)
 

Variables

 VT_TYPE = 4
 
 VT_IN = 6
 
 VT_OUT = 8
 
 VT_DIM_ARG = 10
 
 VT_DIM = 8
 
 VT_KEEP_DIM = 10
 
 VT_USE_MULTICORE = 12
 
 VT_ALL_DIMENSIONS = 8
 

Variable Documentation

◆ VT_ALL_DIMENSIONS

VT_ALL_DIMENSIONS = 8

◆ VT_DIM

VT_DIM = 8

◆ VT_DIM_ARG

VT_DIM_ARG = 10

◆ VT_IN

VT_IN = 6

◆ VT_KEEP_DIM

VT_KEEP_DIM = 10

◆ VT_OUT

VT_OUT = 8

◆ VT_TYPE

VT_TYPE = 4

◆ VT_USE_MULTICORE

VT_USE_MULTICORE = 12