TT-MLIR
Classes | Public Member Functions | List of all members
mlir::tt::DeviceLayoutInterface Class Reference
Inheritance diagram for mlir::tt::DeviceLayoutInterface:

Classes

struct  Trait
 

Public Member Functions

llvm::ArrayRef< int64_t > getGridShape (ShapedType shapedType) const
 Get the grid shape part of the memref shape (aka the first half) More...
 
llvm::ArrayRef< int64_t > getShardShape (ShapedType shapedType) const
 Get the shard shape part of the memref shape (aka the second half) More...
 
int64_t getShardNumElements (ShapedType shapedType) const
 Get the number of elements for the shard shape part of the memref shape (aka the second half) More...
 

Member Function Documentation

◆ getGridShape()

llvm::ArrayRef< int64_t > mlir::tt::DeviceLayoutInterface::getGridShape ( ShapedType  shapedType) const

Get the grid shape part of the memref shape (aka the first half)

◆ getShardNumElements()

int64_t mlir::tt::DeviceLayoutInterface::getShardNumElements ( ShapedType  shapedType) const

Get the number of elements for the shard shape part of the memref shape (aka the second half)

◆ getShardShape()

llvm::ArrayRef< int64_t > mlir::tt::DeviceLayoutInterface::getShardShape ( ShapedType  shapedType) const

Get the shard shape part of the memref shape (aka the second half)


The documentation for this class was generated from the following files: