TT-MLIR
Classes | Namespaces | Functions | Variables
embedding_generated.h File Reference
#include "flatbuffers/flatbuffers.h"
#include "ttmlir/Target/Common/types_generated.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::EmbeddingOpBuilder
 
struct  tt::target::ttnn::EmbeddingOp::Traits
 
struct  tt::target::ttnn::EmbeddingBackwardOpBuilder
 
struct  tt::target::ttnn::EmbeddingBackwardOp::Traits
 

Namespaces

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

Functions

inline ::flatbuffers::Offset< EmbeddingOp > tt::target::ttnn::CreateEmbeddingOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > input=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > weight=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > out=0)
 
inline ::flatbuffers::Offset< EmbeddingBackwardOp > tt::target::ttnn::CreateEmbeddingBackwardOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > input=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > weight=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > in_grad=0, ::flatbuffers::Optional< tt::target::DataType > dtype=::flatbuffers::nullopt, ::flatbuffers::Offset< tt::target::ttnn::MemoryConfig > memcfg=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > out=0)
 

Variables

 VT_INPUT = 4
 
 VT_WEIGHT = 6
 
 VT_IN_GRAD = 8
 
 VT_DTYPE = 10
 
 VT_MEMCFG = 12
 

Variable Documentation

◆ VT_DTYPE

VT_DTYPE = 10

◆ VT_IN_GRAD

VT_IN_GRAD = 8

◆ VT_INPUT

VT_INPUT = 4

◆ VT_MEMCFG

VT_MEMCFG = 12

◆ VT_WEIGHT

VT_WEIGHT = 6