|
using | KeyTy = std::tuple< std::optional< DataType >, std::optional< DataType >, StringAttr, BoolAttr, BoolAttr, std::optional< uint32_t >, std::optional< uint32_t >, BoolAttr, BoolAttr, std::optional< TensorMemoryLayout >, CoreRangeSetAttr, BoolAttr, std::optional< Layout >, BoolAttr, BoolAttr, BoolAttr, BoolAttr, BoolAttr, BoolAttr > |
|
|
| Conv2dConfigAttrStorage (std::optional< DataType > dtype, std::optional< DataType > weights_dtype, StringAttr activation, BoolAttr deallocate_activation, BoolAttr reallocate_halo_output, std::optional< uint32_t > act_block_h_override, std::optional< uint32_t > act_block_w_div, BoolAttr reshard_if_not_optimal, BoolAttr override_sharding_config, std::optional< TensorMemoryLayout > shard_layout, CoreRangeSetAttr core_grid, BoolAttr transpose_shards, std::optional< Layout > output_layout, BoolAttr preprocess_weights_on_device, BoolAttr always_preprocess_weights, BoolAttr enable_act_double_buffer, BoolAttr enable_weights_double_buffer, BoolAttr enable_split_reader, BoolAttr enable_subblock_padding) |
|
KeyTy | getAsKey () const |
|
bool | operator== (const KeyTy &tblgenKey) const |
|
◆ KeyTy
using mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::KeyTy = std::tuple<std::optional<DataType>, std::optional<DataType>, StringAttr, BoolAttr, BoolAttr, std::optional<uint32_t>, std::optional<uint32_t>, BoolAttr, BoolAttr, std::optional<TensorMemoryLayout>, CoreRangeSetAttr, BoolAttr, std::optional<Layout>, BoolAttr, BoolAttr, BoolAttr, BoolAttr, BoolAttr, BoolAttr> |
◆ Conv2dConfigAttrStorage()
mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::Conv2dConfigAttrStorage |
( |
std::optional< DataType > |
dtype, |
|
|
std::optional< DataType > |
weights_dtype, |
|
|
StringAttr |
activation, |
|
|
BoolAttr |
deallocate_activation, |
|
|
BoolAttr |
reallocate_halo_output, |
|
|
std::optional< uint32_t > |
act_block_h_override, |
|
|
std::optional< uint32_t > |
act_block_w_div, |
|
|
BoolAttr |
reshard_if_not_optimal, |
|
|
BoolAttr |
override_sharding_config, |
|
|
std::optional< TensorMemoryLayout > |
shard_layout, |
|
|
CoreRangeSetAttr |
core_grid, |
|
|
BoolAttr |
transpose_shards, |
|
|
std::optional< Layout > |
output_layout, |
|
|
BoolAttr |
preprocess_weights_on_device, |
|
|
BoolAttr |
always_preprocess_weights, |
|
|
BoolAttr |
enable_act_double_buffer, |
|
|
BoolAttr |
enable_weights_double_buffer, |
|
|
BoolAttr |
enable_split_reader, |
|
|
BoolAttr |
enable_subblock_padding |
|
) |
| |
|
inline |
◆ construct()
static Conv2dConfigAttrStorage* mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::construct |
( |
::mlir::AttributeStorageAllocator & |
allocator, |
|
|
KeyTy && |
tblgenKey |
|
) |
| |
|
inlinestatic |
◆ getAsKey()
KeyTy mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::getAsKey |
( |
| ) |
const |
|
inline |
◆ hashKey()
::llvm::hash_code mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::hashKey |
( |
const KeyTy & |
tblgenKey | ) |
|
|
inlinestatic |
◆ operator==()
bool mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::operator== |
( |
const KeyTy & |
tblgenKey | ) |
const |
|
inline |
◆ act_block_h_override
std::optional<uint32_t> mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::act_block_h_override |
◆ act_block_w_div
std::optional<uint32_t> mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::act_block_w_div |
◆ activation
StringAttr mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::activation |
◆ always_preprocess_weights
BoolAttr mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::always_preprocess_weights |
◆ core_grid
◆ deallocate_activation
BoolAttr mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::deallocate_activation |
◆ dtype
std::optional<DataType> mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::dtype |
◆ enable_act_double_buffer
BoolAttr mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::enable_act_double_buffer |
◆ enable_split_reader
BoolAttr mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::enable_split_reader |
◆ enable_subblock_padding
BoolAttr mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::enable_subblock_padding |
◆ enable_weights_double_buffer
BoolAttr mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::enable_weights_double_buffer |
◆ output_layout
std::optional<Layout> mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::output_layout |
◆ override_sharding_config
BoolAttr mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::override_sharding_config |
◆ preprocess_weights_on_device
BoolAttr mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::preprocess_weights_on_device |
◆ reallocate_halo_output
BoolAttr mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::reallocate_halo_output |
◆ reshard_if_not_optimal
BoolAttr mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::reshard_if_not_optimal |
◆ shard_layout
std::optional<TensorMemoryLayout> mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::shard_layout |
◆ transpose_shards
BoolAttr mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::transpose_shards |
◆ weights_dtype
std::optional<DataType> mlir::tt::ttnn::detail::Conv2dConfigAttrStorage::weights_dtype |
The documentation for this struct was generated from the following file: