mlir::shape Namespace Reference

Classes
struct	ShapeMappingAnalysis
	ShapeMappingAnalysis is used together with OutlineShapeComputationPass to preserve Value and corresponding shape function / arguments mapping information. More...
struct	ShapeMappingValue
	ShapeMappingValue works as the value of ShapeMappingAnalysis table, where funcSymbol is the symbol of mapping function, and inputs are the actual parameters for the function. More...

Functions
RankedTensorType	getExtentTensorType (MLIRContext *ctx, int64_t rank=ShapedType::kDynamic)
	Alias type for extent tensors.
bool	isExtentTensorType (Type)
LogicalResult	getShapeVec (Value input, SmallVectorImpl< int64_t > &shapeValues)
void	registerBufferizableOpInterfaceExternalModels (DialectRegistry &registry)

Function Documentation

getExtentTensorType()

RankedTensorType mlir::shape::getExtentTensorType	(	MLIRContext *	ctx,
		int64_t	rank = ShapedType::kDynamic )

Alias type for extent tensors.

Definition at line 40 of file Shape.cpp.

Referenced by mlir::quant::createLowerQuantOps().

getShapeVec()

LogicalResult mlir::shape::getShapeVec	(	Value	input,
		SmallVectorImpl< int64_t > &	shapeValues )

Definition at line 49 of file Shape.cpp.

References mlir::Value::getDefiningOp(), mlir::m_Constant(), mlir::matchPattern(), and success().

isExtentTensorType()

bool mlir::shape::isExtentTensorType

(

Type

type

)

Definition at line 44 of file Shape.cpp.

registerBufferizableOpInterfaceExternalModels()

void mlir::shape::registerBufferizableOpInterfaceExternalModels

(

DialectRegistry &

registry

)

Definition at line 139 of file BufferizableOpInterfaceImpl.cpp.

References mlir::DialectRegistry::addExtension().

Referenced by mlir::registerAllDialects().