TT-MLIR
|
This is the complete list of members for mlir::tt::MetalLayoutAttr, including all inherited members.
dialectName | mlir::tt::MetalLayoutAttr | static |
get(::mlir::MLIRContext *context, ArrayRef< int64_t > tensorShape, Type elementType, MemorySpace memorySpace=MemorySpace::System, GridAttr grid={}, ArrayRef< std::pair< std::int64_t, std::int64_t >> collapseIntervals={{0, -1}}, OOBVal oobVal=OOBVal::Undef, TensorMemoryLayout memLayout=TensorMemoryLayout::None) | mlir::tt::MetalLayoutAttr | static |
get(::mlir::MLIRContext *context, RankedTensorType ty, MemorySpace memorySpace=MemorySpace::System, GridAttr grid={}, ArrayRef< std::pair< std::int64_t, std::int64_t >> collapseIntervals={{0, -1}}, OOBVal oobVal=OOBVal::Undef, TensorMemoryLayout memLayout=TensorMemoryLayout::None) | mlir::tt::MetalLayoutAttr | static |
get(::mlir::MLIRContext *context, RankedTensorType ty, MemorySpace memorySpace, GridAttr grid, Type elementType, TensorMemoryLayout memLayout=TensorMemoryLayout::None) | mlir::tt::MetalLayoutAttr | static |
get(::mlir::MLIRContext *context, AffineMap linear, OOBVal oob_val, GridAttr grid, MemRefType memref, TensorMemoryLayout mem_layout) | mlir::tt::MetalLayoutAttr | static |
getElementSizeBytes() const | mlir::tt::MetalLayoutAttr | |
getElementType() const | mlir::tt::MetalLayoutAttr | |
getGrid() const | mlir::tt::MetalLayoutAttr | |
getIdentityTileLinearMap() const | mlir::tt::MetalLayoutAttr | |
getLinear() const | mlir::tt::MetalLayoutAttr | |
getMemLayout() const | mlir::tt::MetalLayoutAttr | |
getMemorySpace() const | mlir::tt::MetalLayoutAttr | |
getMemref() const | mlir::tt::MetalLayoutAttr | |
getMemrefSizeBytes() const | mlir::tt::MetalLayoutAttr | |
getMnemonic() | mlir::tt::MetalLayoutAttr | inlinestatic |
getOobVal() const | mlir::tt::MetalLayoutAttr | |
getPhysicalShape(ArrayRef< int64_t > logicalShape) const | mlir::tt::MetalLayoutAttr | |
getScalarElementType() const | mlir::tt::MetalLayoutAttr | |
getShardShape(bool convertTileToScalar=true) const | mlir::tt::MetalLayoutAttr | |
getStride(ArrayRef< int64_t > logicalShape) const | mlir::tt::MetalLayoutAttr | |
getTiledShape(ArrayRef< int64_t > logicalTensorShape) const | mlir::tt::MetalLayoutAttr | |
hasInterleavedL1TensorMemoryLayout() const | mlir::tt::MetalLayoutAttr | |
hasInterleavedTensorMemoryLayout() const | mlir::tt::MetalLayoutAttr | |
hasShardedL1TensorMemoryLayout() const | mlir::tt::MetalLayoutAttr | |
hasShardedTensorMemoryLayout() const | mlir::tt::MetalLayoutAttr | |
isDeviceMemorySpace() const | mlir::tt::MetalLayoutAttr | inline |
isSystemMemorySpace() const | mlir::tt::MetalLayoutAttr | inline |
isTiled() const | mlir::tt::MetalLayoutAttr | |
name | mlir::tt::MetalLayoutAttr | static |
parse(::mlir::AsmParser &odsParser, ::mlir::Type odsType) | mlir::tt::MetalLayoutAttr | static |
print(::mlir::AsmPrinter &odsPrinter) const | mlir::tt::MetalLayoutAttr | |
projectOnto(AffineMap linearMap, AffineMap physicalMemoryMap) const | mlir::tt::MetalLayoutAttr | |
replaceMemoryMapSymbolsWithShardShape(AffineMap physicalMemoryMap) const | mlir::tt::MetalLayoutAttr | |
withElementType(::mlir::MLIRContext *context, Type elementType) | mlir::tt::MetalLayoutAttr | |
withGrid(::mlir::MLIRContext *context, ArrayRef< int64_t > tensorShape, GridAttr grid, ArrayRef< std::pair< std::int64_t, std::int64_t >> collapseIntervals={{0, -1}}) | mlir::tt::MetalLayoutAttr | |
withGrid(::mlir::MLIRContext *context, RankedTensorType ty, GridAttr grid, ArrayRef< std::pair< std::int64_t, std::int64_t >> collapseIntervals={{0, -1}}) | mlir::tt::MetalLayoutAttr | |
withMemoryLayout(::mlir::MLIRContext *context, TensorMemoryLayout memLayout) | mlir::tt::MetalLayoutAttr | |
withMemorySpace(::mlir::MLIRContext *context, MemorySpace memorySpace) | mlir::tt::MetalLayoutAttr | |
withShardShape(::mlir::MLIRContext *context, llvm::SmallVector< int64_t > shardShape) | mlir::tt::MetalLayoutAttr |