pool_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::Scale2DTraits< T >
struct	tt::target::ttnn::Scale2DTraits< tt::target::ttnn::UniformScale2D >
struct	tt::target::ttnn::Scale2DTraits< tt::target::ttnn::NonUniformScale2D >
struct	tt::target::ttnn::FLATBUFFERS_FINAL_CLASS
struct	tt::target::ttnn::MaxPool2dOpBuilder
struct	tt::target::ttnn::MaxPool2dOp::Traits
struct	tt::target::ttnn::UniformScale2DBuilder
struct	tt::target::ttnn::UniformScale2D::Traits
struct	tt::target::ttnn::NonUniformScale2DBuilder
struct	tt::target::ttnn::NonUniformScale2D::Traits
struct	tt::target::ttnn::UpsampleOpBuilder
struct	tt::target::ttnn::UpsampleOp::Traits

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

Enumerations
enum class	tt::target::ttnn::Scale2D : uint8_t {   tt::target::ttnn::NONE = 0 , tt::target::ttnn::UniformScale2D = 1 , tt::target::ttnn::NonUniformScale2D = 2 , tt::target::ttnn::MIN = NONE ,   tt::target::ttnn::MAX = NonUniformScale2D }

Functions
const Scale2D(&	tt::target::ttnn::EnumValuesScale2D ())[3]
const char *const *	tt::target::ttnn::EnumNamesScale2D ()
const char *	tt::target::ttnn::EnumNameScale2D (Scale2D e)
bool	tt::target::ttnn::VerifyScale2D (::flatbuffers::Verifier &verifier, const void *obj, Scale2D type)
bool	tt::target::ttnn::VerifyScale2DVector (::flatbuffers::Verifier &verifier, const ::flatbuffers::Vector<::flatbuffers::Offset< void >> *values, const ::flatbuffers::Vector< Scale2D > *types)
inline ::flatbuffers::Offset< MaxPool2dOp >	tt::target::ttnn::CreateMaxPool2dOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > in=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > out=0, uint32_t batch_size=0, uint32_t input_height=0, uint32_t input_width=0, uint32_t channels=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 >> dilation=0, bool ceil_mode=false)
inline ::flatbuffers::Offset< MaxPool2dOp >	tt::target::ttnn::CreateMaxPool2dOpDirect (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > in=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > out=0, uint32_t batch_size=0, uint32_t input_height=0, uint32_t input_width=0, uint32_t channels=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 > *dilation=nullptr, bool ceil_mode=false)
inline ::flatbuffers::Offset< UniformScale2D >	tt::target::ttnn::CreateUniformScale2D (::flatbuffers::FlatBufferBuilder &_fbb, int32_t scale=0)
inline ::flatbuffers::Offset< NonUniformScale2D >	tt::target::ttnn::CreateNonUniformScale2D (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset<::flatbuffers::Vector< int32_t >> scale=0)
inline ::flatbuffers::Offset< NonUniformScale2D >	tt::target::ttnn::CreateNonUniformScale2DDirect (::flatbuffers::FlatBufferBuilder &_fbb, const std::vector< int32_t > *scale=nullptr)
template<>
const tt::target::ttnn::UniformScale2D *	tt::target::ttnn::UpsampleOp::scale_factor_as< tt::target::ttnn::UniformScale2D > () const
template<>
const tt::target::ttnn::NonUniformScale2D *	tt::target::ttnn::UpsampleOp::scale_factor_as< tt::target::ttnn::NonUniformScale2D > () const
inline ::flatbuffers::Offset< UpsampleOp >	tt::target::ttnn::CreateUpsampleOp (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > in=0, tt::target::ttnn::Scale2D scale_factor_type=tt::target::ttnn::Scale2D::NONE, ::flatbuffers::Offset< void > scale_factor=0, ::flatbuffers::Offset<::flatbuffers::String > mode=0, ::flatbuffers::Offset< tt::target::ttnn::MemoryConfig > memory_config=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > out=0)
inline ::flatbuffers::Offset< UpsampleOp >	tt::target::ttnn::CreateUpsampleOpDirect (::flatbuffers::FlatBufferBuilder &_fbb, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > in=0, tt::target::ttnn::Scale2D scale_factor_type=tt::target::ttnn::Scale2D::NONE, ::flatbuffers::Offset< void > scale_factor=0, const char *mode=nullptr, ::flatbuffers::Offset< tt::target::ttnn::MemoryConfig > memory_config=0, ::flatbuffers::Offset< tt::target::ttnn::TensorRef > out=0)

Variables
	VT_IN = 4
	VT_OUT = 6
	VT_BATCH_SIZE = 8
	VT_INPUT_HEIGHT = 10
	VT_INPUT_WIDTH = 12
	VT_CHANNELS = 14
	VT_KERNEL_SIZE = 16
	VT_STRIDE = 18
	VT_PADDING = 20
	VT_DILATION = 22
	VT_SCALE_FACTOR_TYPE = 6
	VT_SCALE_FACTOR = 8
	VT_MODE = 10
	VT_MEMORY_CONFIG = 12

Variable Documentation

VT_BATCH_SIZE

VT_BATCH_SIZE = 8

VT_CHANNELS

VT_CHANNELS = 14

VT_DILATION

VT_DILATION = 22

VT_IN

VT_IN = 4

VT_INPUT_HEIGHT

VT_INPUT_HEIGHT = 10

VT_INPUT_WIDTH

VT_INPUT_WIDTH = 12

VT_KERNEL_SIZE

VT_KERNEL_SIZE = 16

VT_MEMORY_CONFIG

VT_MEMORY_CONFIG = 12

VT_MODE

VT_MODE = 10

VT_OUT

VT_OUT = 6

VT_PADDING

VT_PADDING = 20

VT_SCALE_FACTOR

VT_SCALE_FACTOR = 8

VT_SCALE_FACTOR_TYPE

VT_SCALE_FACTOR_TYPE = 6

VT_STRIDE

VT_STRIDE = 18