TT-MLIR
|
This is the complete list of members for mlir::tt::ttnn::TTNNLayoutAttr, including all inherited members.
calculateLogicalShardShapeForL1Interleaved(ArrayRef< int64_t > tensorShape, Type elementType, mlir::AffineMap linear, GridAttr grid) | mlir::tt::ttnn::TTNNLayoutAttr | static |
calculateLogicalShardShapeForSharding(ArrayRef< int64_t > tensorShape, mlir::AffineMap linear, GridAttr grid) | mlir::tt::ttnn::TTNNLayoutAttr | static |
dialectName | mlir::tt::ttnn::TTNNLayoutAttr | static |
get(::mlir::MLIRContext *context, ArrayRef< int64_t > tensorShape, Type elementType, BufferType bufferType, GridAttr grid, TensorMemoryLayoutAttr memoryLayoutAttr=nullptr, TensorMeshShardingAttr tensorMeshShardingAttr=nullptr, ArrayRef< std::pair< std::int64_t, std::int64_t >> collapseIntervals={{0, -1}}) | mlir::tt::ttnn::TTNNLayoutAttr | static |
get(::mlir::MLIRContext *context, AffineMap linear, GridAttr grid, MemRefType memref, TensorMemoryLayoutAttr mem_layout, TensorMeshShardingAttr tensor_mesh_sharding) | mlir::tt::ttnn::TTNNLayoutAttr | static |
getBufferType() const | mlir::tt::ttnn::TTNNLayoutAttr | |
getChecked(::llvm::function_ref<::mlir::InFlightDiagnostic()> emitError, ::mlir::MLIRContext *context, AffineMap linear, GridAttr grid, MemRefType memref, TensorMemoryLayoutAttr mem_layout, TensorMeshShardingAttr tensor_mesh_sharding) | mlir::tt::ttnn::TTNNLayoutAttr | static |
getDataType() const | mlir::tt::ttnn::TTNNLayoutAttr | |
getElementSizeBytes() const | mlir::tt::ttnn::TTNNLayoutAttr | |
getElementType() const | mlir::tt::ttnn::TTNNLayoutAttr | |
getGrid() const | mlir::tt::ttnn::TTNNLayoutAttr | |
getIdentityTileLinearMap() const | mlir::tt::ttnn::TTNNLayoutAttr | |
getLayout() const | mlir::tt::ttnn::TTNNLayoutAttr | |
getLinear() const | mlir::tt::ttnn::TTNNLayoutAttr | |
getMemLayout() const | mlir::tt::ttnn::TTNNLayoutAttr | |
getMemLayoutOpt() const | mlir::tt::ttnn::TTNNLayoutAttr | |
getMemref() const | mlir::tt::ttnn::TTNNLayoutAttr | |
getMnemonic() | mlir::tt::ttnn::TTNNLayoutAttr | inlinestatic |
getScalarElementType() const | mlir::tt::ttnn::TTNNLayoutAttr | |
getScalarShardShape() const | mlir::tt::ttnn::TTNNLayoutAttr | |
getShardShape() const | mlir::tt::ttnn::TTNNLayoutAttr | |
getShardSizeInBytes() const | mlir::tt::ttnn::TTNNLayoutAttr | |
getTensorMeshSharding() const | mlir::tt::ttnn::TTNNLayoutAttr | |
getTiledShape(ArrayRef< int64_t > logicalTensorShape) const | mlir::tt::ttnn::TTNNLayoutAttr | |
hasDRAMBufferType() const | mlir::tt::ttnn::TTNNLayoutAttr | |
hasInterleavedDRAMTensorMemoryLayout() const | mlir::tt::ttnn::TTNNLayoutAttr | |
hasInterleavedL1TensorMemoryLayout() const | mlir::tt::ttnn::TTNNLayoutAttr | |
hasL1BufferType() const | mlir::tt::ttnn::TTNNLayoutAttr | |
hasShardedL1TensorMemoryLayout() const | mlir::tt::ttnn::TTNNLayoutAttr | |
hasShardedTensorMemoryLayout() const | mlir::tt::ttnn::TTNNLayoutAttr | |
isDeviceBufferType() const | mlir::tt::ttnn::TTNNLayoutAttr | inline |
isMeshDeviceTensor() const | mlir::tt::ttnn::TTNNLayoutAttr | inline |
isSystemBufferType() const | mlir::tt::ttnn::TTNNLayoutAttr | inline |
isTiled() const | mlir::tt::ttnn::TTNNLayoutAttr | |
name | mlir::tt::ttnn::TTNNLayoutAttr | static |
parse(::mlir::AsmParser &odsParser, ::mlir::Type odsType) | mlir::tt::ttnn::TTNNLayoutAttr | static |
print(::mlir::AsmPrinter &odsPrinter) const | mlir::tt::ttnn::TTNNLayoutAttr | |
replaceMemoryMapSymbolsWithShardShape(AffineMap physicalMemoryMap) const | mlir::tt::ttnn::TTNNLayoutAttr | |
verify(::llvm::function_ref<::mlir::InFlightDiagnostic()> emitError, AffineMap linear, GridAttr grid, MemRefType memref, TensorMemoryLayoutAttr mem_layout, TensorMeshShardingAttr tensor_mesh_sharding) | mlir::tt::ttnn::TTNNLayoutAttr | static |
verifyInvariants(::llvm::function_ref<::mlir::InFlightDiagnostic()> emitError, AffineMap linear, GridAttr grid, MemRefType memref, TensorMemoryLayoutAttr mem_layout, TensorMeshShardingAttr tensor_mesh_sharding) | mlir::tt::ttnn::TTNNLayoutAttr | static |
withBufferType(::mlir::MLIRContext *context, BufferType bufferType) | mlir::tt::ttnn::TTNNLayoutAttr | |
withElementType(::mlir::MLIRContext *context, Type elementType, ArrayRef< int64_t > tensorShape, ArrayRef< std::pair< std::int64_t, std::int64_t >> collapseIntervals={{0, -1}}) | mlir::tt::ttnn::TTNNLayoutAttr | |
withGrid(::mlir::MLIRContext *context, ArrayRef< int64_t > tensorShape, GridAttr grid, ArrayRef< std::pair< std::int64_t, std::int64_t >> collapseIntervals={{0, -1}}) | mlir::tt::ttnn::TTNNLayoutAttr | |
withGrid(::mlir::MLIRContext *context, RankedTensorType ty, GridAttr grid, ArrayRef< std::pair< std::int64_t, std::int64_t >> collapseIntervals={{0, -1}}) | mlir::tt::ttnn::TTNNLayoutAttr | |
withMemoryLayout(::mlir::MLIRContext *context, TensorMemoryLayoutAttr memLayoutAttr) | mlir::tt::ttnn::TTNNLayoutAttr | |
withMemoryLayout(::mlir::MLIRContext *context, TensorMemoryLayout memLayout) | mlir::tt::ttnn::TTNNLayoutAttr | |
withShardShape(::mlir::MLIRContext *context, llvm::SmallVector< int64_t > shardShape) | mlir::tt::ttnn::TTNNLayoutAttr | |
withTensorShape(::mlir::MLIRContext *context, ArrayRef< int64_t > tensorShape) | mlir::tt::ttnn::TTNNLayoutAttr |