SPIR-V TensorARM Type. More...

#include "mlir/Dialect/SPIRV/IR/SPIRVTypes.h"

Inheritance diagram for mlir::spirv::TensorArmType:

Public Types
using	ShapedTypeTraits = ShapedType::Trait< TensorArmType >

using	Base = StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits... >
	Utility declarations for the concrete attribute class. More...

Public Types inherited from mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >
using	Base = StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits... >
	Utility declarations for the concrete attribute class. More...

using	ImplType = StorageT

using	HasTraitFn = bool(*)(TypeID)

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

void	getExtensions (SPIRVType::ExtensionArrayRefVector &extensions, std::optional< StorageClass > storage=std::nullopt)

void	getCapabilities (SPIRVType::CapabilityArrayRefVector &capabilities, std::optional< StorageClass > storage=std::nullopt)

Public Member Functions inherited from mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >
ImplType *	getImpl () const
	Utility for easy access to the storage instance. More...

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< ConcreteT, BaseT, StorageT, UniquerT, Traits >
static TypeID	getTypeID ()
	Return a unique identifier for the concrete type. More...

template<typename T >
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. More...

static detail::InterfaceMap	getInterfaceMap ()
	Returns an interface map for the interfaces registered to this storage user. More...

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. More...

static auto	getWalkImmediateSubElementsFn ()
	Returns a function that walks immediate sub elements of a given instance of the storage user. More...

static auto	getReplaceImmediateSubElementsFn ()
	Returns a function that replaces immediate sub elements of a given instance of the storage user. More...

template<typename... IfaceModels>
static void	attachInterface (MLIRContext &context)
	Attach the given models as implementations of the corresponding interfaces for the concrete storage user class. More...

template<typename... Args>
static ConcreteT	get (MLIRContext *ctx, Args &&...args)
	Get or create a new ConcreteT instance within the ctx. More...

template<typename... Args>
static ConcreteT	getChecked (const Location &loc, Args &&...args)
	Get or create a new ConcreteT instance within the ctx, defined at the given, potentially unknown, location. More...

template<typename... Args>
static ConcreteT	getChecked (function_ref< InFlightDiagnostic()> emitErrorFn, MLIRContext *ctx, Args... args)
	Get or create a new ConcreteT instance within the ctx. More...

static ConcreteT	getFromOpaquePointer (const void *ptr)
	Get an instance of the concrete type from a void pointer. More...

Static Public Attributes
static constexpr StringLiteral	name = "spirv.arm.tensor"

Additional Inherited Members
Protected Member Functions inherited from mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >
template<typename... Args>
LogicalResult	mutate (Args &&...args)
	Mutate the current storage instance. More...

Static Protected Member Functions inherited from mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >
template<typename... Args>
static LogicalResult	verifyInvariants (Args... args)
	Default implementation that just returns success. More...

Detailed Description

SPIR-V TensorARM Type.

Definition at line 522 of file SPIRVTypes.h.

Member Typedef Documentation

◆ Base

using mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >::Base = StorageUserBase<ConcreteT, BaseT, StorageT, UniquerT, Traits...>

Utility declarations for the concrete attribute class.

Definition at line 100 of file StorageUniquerSupport.h.

◆ ShapedTypeTraits

using mlir::spirv::TensorArmType::ShapedTypeTraits = ShapedType::Trait<TensorArmType>

Definition at line 528 of file SPIRVTypes.h.

Member Function Documentation

◆ cloneWith()

TensorArmType TensorArmType::cloneWith	(	std::optional< ArrayRef< int64_t >>	shape,
		Type	elementType
	)		const

Definition at line 1335 of file SPIRVTypes.cpp.

References get(), and getShape().

◆ get()

TensorArmType TensorArmType::get	(	ArrayRef< int64_t >	shape,
		Type	elementType
	)

static

Definition at line 1331 of file SPIRVTypes.cpp.

References mlir::get(), and mlir::Type::getContext().

Referenced by cloneWith(), and parseTensorArmType().

◆ getCapabilities()

void TensorArmType::getCapabilities	(	SPIRVType::CapabilityArrayRefVector &	capabilities,
		std::optional< StorageClass >	storage = `std::nullopt`
	)

Definition at line 1352 of file SPIRVTypes.cpp.

References getElementType().

◆ getElementType()

Type TensorArmType::getElementType ( ) const

Definition at line 1340 of file SPIRVTypes.cpp.

References mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >::getImpl().

Referenced by mlir::spirv::CompositeType::getCapabilities(), getCapabilities(), mlir::spirv::CompositeType::getExtensions(), getExtensions(), and print().

◆ getExtensions()

void TensorArmType::getExtensions	(	SPIRVType::ExtensionArrayRefVector &	extensions,
		std::optional< StorageClass >	storage = `std::nullopt`
	)

Definition at line 1343 of file SPIRVTypes.cpp.

References getElementType().

◆ getShape()

ArrayRef< int64_t > TensorArmType::getShape ( ) const

Definition at line 1341 of file SPIRVTypes.cpp.

References mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >::getImpl().

Referenced by cloneWith(), hasRank(), and print().

◆ hasRank()

bool mlir::spirv::TensorArmType::hasRank ( ) const

inline

Definition at line 552 of file SPIRVTypes.h.

References getShape().

Referenced by print().

◆ operator ShapedType()

mlir::spirv::TensorArmType::operator ShapedType ( ) const

inline

Definition at line 553 of file SPIRVTypes.h.

◆ verifyInvariants()

LogicalResult TensorArmType::verifyInvariants	(	function_ref< InFlightDiagnostic()>	emitError,
		ArrayRef< int64_t >	shape,
		Type	elementType
	)

static

Definition at line 1362 of file SPIRVTypes.cpp.

References mlir::emitError().

Member Data Documentation

◆ name

constexpr StringLiteral mlir::spirv::TensorArmType::name = "spirv.arm.tensor"

staticconstexpr

Definition at line 538 of file SPIRVTypes.h.

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

include/mlir/Dialect/SPIRV/IR/SPIRVTypes.h
lib/Dialect/SPIRV/IR/SPIRVTypes.cpp

Public Types

Public Member Functions

Static Public Member Functions

Static Public Attributes

Additional Inherited Members