TT-MLIR
Public Types | Public Member Functions | Static Public Member Functions | Public Attributes | List of all members
mlir::tt::detail::TensorMeshShardingAxisAttrStorage Struct Reference
Inheritance diagram for mlir::tt::detail::TensorMeshShardingAxisAttrStorage:

Public Types

using KeyTy = std::tuple< int64_t, ::llvm::ArrayRef< int64_t > >
 

Public Member Functions

 TensorMeshShardingAxisAttrStorage (int64_t shard_shape, ::llvm::ArrayRef< int64_t > axes)
 
KeyTy getAsKey () const
 
bool operator== (const KeyTy &tblgenKey) const
 

Static Public Member Functions

::llvm::hash_code hashKey (const KeyTy &tblgenKey)
 
static TensorMeshShardingAxisAttrStorageconstruct (::mlir::AttributeStorageAllocator &allocator, KeyTy &&tblgenKey)
 

Public Attributes

int64_t shard_shape
 
::llvm::ArrayRef< int64_t > axes
 

Member Typedef Documentation

◆ KeyTy

using mlir::tt::detail::TensorMeshShardingAxisAttrStorage::KeyTy = std::tuple<int64_t, ::llvm::ArrayRef<int64_t> >

Constructor & Destructor Documentation

◆ TensorMeshShardingAxisAttrStorage()

mlir::tt::detail::TensorMeshShardingAxisAttrStorage::TensorMeshShardingAxisAttrStorage ( int64_t  shard_shape,
::llvm::ArrayRef< int64_t >  axes 
)
inline

Member Function Documentation

◆ construct()

static TensorMeshShardingAxisAttrStorage* mlir::tt::detail::TensorMeshShardingAxisAttrStorage::construct ( ::mlir::AttributeStorageAllocator &  allocator,
KeyTy &&  tblgenKey 
)
inlinestatic

◆ getAsKey()

KeyTy mlir::tt::detail::TensorMeshShardingAxisAttrStorage::getAsKey ( ) const
inline

◆ hashKey()

::llvm::hash_code mlir::tt::detail::TensorMeshShardingAxisAttrStorage::hashKey ( const KeyTy tblgenKey)
inlinestatic

◆ operator==()

bool mlir::tt::detail::TensorMeshShardingAxisAttrStorage::operator== ( const KeyTy tblgenKey) const
inline

Member Data Documentation

◆ axes

::llvm::ArrayRef<int64_t> mlir::tt::detail::TensorMeshShardingAxisAttrStorage::axes

◆ shard_shape

int64_t mlir::tt::detail::TensorMeshShardingAxisAttrStorage::shard_shape

The documentation for this struct was generated from the following file: