|
MLIR 22.0.0git
|
SPIR-V TensorARM Type. More...
#include "mlir/Dialect/SPIRV/IR/SPIRVTypes.h"
Public Types | |
| using | ShapedTypeTraits = ShapedType::Trait<TensorArmType> |
| Public Types inherited from mlir::detail::StorageUserBase< ConcreteType, BaseType, StorageType, detail::TypeUniquer, Traits... > | |
| using | Base |
| Utility declarations for the concrete attribute class. | |
| using | ImplType |
| using | HasTraitFn |
Public Member Functions | |
| TensorArmType | cloneWith (std::optional< ArrayRef< int64_t > > shape, Type elementType) const |
| Type | getElementType () const |
| ArrayRef< int64_t > | getShape () const |
| bool | hasRank () const |
| operator ShapedType () const | |
| Public Member Functions inherited from mlir::detail::StorageUserBase< ConcreteType, BaseType, StorageType, detail::TypeUniquer, Traits... > | |
| ImplType * | getImpl () const |
| Utility for easy access to the storage instance. | |
Static Public Member Functions | |
| static TensorArmType | get (ArrayRef< int64_t > shape, Type elementType) |
| static LogicalResult | verifyInvariants (function_ref< InFlightDiagnostic()> emitError, ArrayRef< int64_t > shape, Type elementType) |
| Static Public Member Functions inherited from mlir::detail::StorageUserBase< ConcreteType, BaseType, StorageType, detail::TypeUniquer, Traits... > | |
| static TypeID | getTypeID () |
| Return a unique identifier for the concrete type. | |
| static bool | classof (T val) |
| Provide an implementation of 'classof' that compares the type id of the provided value with that of the concrete type. | |
| static detail::InterfaceMap | getInterfaceMap () |
| Returns an interface map for the interfaces registered to this storage user. | |
| static HasTraitFn | getHasTraitFn () |
| Returns the function that returns true if the given Trait ID matches the IDs of any of the traits defined by the storage user. | |
| static auto | getWalkImmediateSubElementsFn () |
| Returns a function that walks immediate sub elements of a given instance of the storage user. | |
| static auto | getReplaceImmediateSubElementsFn () |
| Returns a function that replaces immediate sub elements of a given instance of the storage user. | |
| static void | attachInterface (MLIRContext &context) |
| Attach the given models as implementations of the corresponding interfaces for the concrete storage user class. | |
| static ConcreteType | get (MLIRContext *ctx, Args &&...args) |
| Get or create a new ConcreteT instance within the ctx. | |
| static ConcreteType | getChecked (const Location &loc, Args &&...args) |
| Get or create a new ConcreteT instance within the ctx, defined at the given, potentially unknown, location. | |
| static ConcreteType | getChecked (function_ref< InFlightDiagnostic()> emitErrorFn, MLIRContext *ctx, Args... args) |
| Get or create a new ConcreteT instance within the ctx. | |
| static ConcreteType | getFromOpaquePointer (const void *ptr) |
| Get an instance of the concrete type from a void pointer. | |
Static Public Attributes | |
| static constexpr StringLiteral | name = "spirv.arm.tensor" |
Additional Inherited Members | |
| Protected Member Functions inherited from mlir::detail::StorageUserBase< ConcreteType, BaseType, StorageType, detail::TypeUniquer, Traits... > | |
| LogicalResult | mutate (Args &&...args) |
| Mutate the current storage instance. | |
| Static Protected Member Functions inherited from mlir::detail::StorageUserBase< ConcreteType, BaseType, StorageType, detail::TypeUniquer, Traits... > | |
| static LogicalResult | verifyInvariants (Args... args) |
| Default implementation that just returns success. | |
SPIR-V TensorARM Type.
Definition at line 463 of file SPIRVTypes.h.
| using mlir::spirv::TensorArmType::ShapedTypeTraits = ShapedType::Trait<TensorArmType> |
Definition at line 469 of file SPIRVTypes.h.
| TensorArmType TensorArmType::cloneWith | ( | std::optional< ArrayRef< int64_t > > | shape, |
| Type | elementType ) const |
Definition at line 1269 of file SPIRVTypes.cpp.
References get(), and getShape().
|
static |
Definition at line 1265 of file SPIRVTypes.cpp.
References mlir::detail::StorageUserBase< ConcreteType, BaseType, StorageType, detail::TypeUniquer, Traits... >::get(), and mlir::Type::getContext().
Referenced by cloneWith(), parseTensorArmType(), and mlir::spirv::Deserializer::processTensorARMType().
| Type TensorArmType::getElementType | ( | ) | const |
Definition at line 1274 of file SPIRVTypes.cpp.
Referenced by print().
Definition at line 1275 of file SPIRVTypes.cpp.
Referenced by cloneWith(), hasRank(), and print().
|
inline |
|
inline |
Definition at line 494 of file SPIRVTypes.h.
|
static |
Definition at line 1290 of file SPIRVTypes.cpp.
References mlir::emitError(), and success().
|
staticconstexpr |
Definition at line 479 of file SPIRVTypes.h.