Python wrapper for InferShapedTypeOpInterface. More...

Inheritance diagram for mlir::python::PyInferShapedTypeOpInterface:

Classes
struct	AppendResultsCallbackData
	C-style user-data structure for type appending callback. More...

Public Member Functions
std::vector< PyShapedTypeComponents >	inferReturnTypeComponents (std::optional< nb::list > operandList, std::optional< PyAttribute > attributes, void *properties, std::optional< std::vector< PyRegion >> regions, DefaultingPyMlirContext context, DefaultingPyLocation location)
	Given the arguments required to build an operation, attempts to infer the shaped type components. More...

Public Member Functions inherited from mlir::python::PyConcreteOpInterface< PyInferShapedTypeOpInterface >
	PyConcreteOpInterface (nb::object object, DefaultingPyMlirContext context)
	Constructs an interface instance from an object that is either an operation or a subclass of OpView. More...

bool	isStatic ()
	Returns `true` if this object was constructed from a subclass of OpView rather than from an operation instance. More...

nb::object	getOperationObject ()
	Returns the operation instance from which this object was constructed. More...

nb::object	getOpView ()
	Returns the opview of the operation instance from which this object was constructed. More...

const std::string &	getOpName ()
	Returns the canonical name of the operation this interface is constructed from. More...

Static Public Member Functions
static void	appendResultsCallback (bool hasRank, intptr_t rank, const int64_t shape, MlirType elementType, MlirAttribute attribute, void userData)
	Appends the shaped type components provided as unpacked shape, element type, attribute to the user-data. More...

static void	bindDerived (ClassTy &cls)

Static Public Member Functions inherited from mlir::python::PyConcreteOpInterface< PyInferShapedTypeOpInterface >
static void	bind (nb::module_ &m)
	Creates the Python bindings for this class in the given module. More...

static void	bindDerived (ClassTy &cls)
	Hook for derived classes to add class-specific bindings. More...

Static Public Attributes
constexpr static const char *	pyClassName = "InferShapedTypeOpInterface"

constexpr static GetTypeIDFunctionTy	getInterfaceID

Additional Inherited Members
Protected Types inherited from mlir::python::PyConcreteOpInterface< PyInferShapedTypeOpInterface >
using	ClassTy = nb::class_< PyInferShapedTypeOpInterface >

using	GetTypeIDFunctionTy = MlirTypeID(*)()

Detailed Description

Python wrapper for InferShapedTypeOpInterface.

This interface has only static methods.

Definition at line 399 of file IRInterfaces.cpp.

Member Function Documentation

◆ appendResultsCallback()

static void mlir::python::PyInferShapedTypeOpInterface::appendResultsCallback	(	bool	hasRank,
		intptr_t	rank,
		const int64_t *	shape,
		MlirType	elementType,
		MlirAttribute	attribute,
		void *	userData
	)

inlinestatic

Appends the shaped type components provided as unpacked shape, element type, attribute to the user-data.

Definition at line 416 of file IRInterfaces.cpp.

References mlir::python::PyInferShapedTypeOpInterface::AppendResultsCallbackData::inferredShapedTypeComponents.

Referenced by inferReturnTypeComponents().

◆ bindDerived()

static void mlir::python::PyInferShapedTypeOpInterface::bindDerived ( ClassTy & cls )

inlinestatic

Definition at line 463 of file IRInterfaces.cpp.

References inferReturnTypeComponents(), and mlir::python::inferReturnTypeComponentsDoc.

◆ inferReturnTypeComponents()

std::vector<PyShapedTypeComponents> mlir::python::PyInferShapedTypeOpInterface::inferReturnTypeComponents	(	std::optional< nb::list >	operandList,
		std::optional< PyAttribute >	attributes,
		void *	properties,
		std::optional< std::vector< PyRegion >>	regions,
		DefaultingPyMlirContext	context,
		DefaultingPyLocation	location
	)

inline

Given the arguments required to build an operation, attempts to infer the shaped type components.

Throws value_error on failure.

Definition at line 434 of file IRInterfaces.cpp.

References appendResultsCallback(), mlir::python::PyMlirContext::get(), mlir::python::PyConcreteOpInterface< PyInferShapedTypeOpInterface >::getOpName(), mlirAttributeGetNull(), mlirInferShapedTypeOpInterfaceInferReturnTypes(), mlirLogicalResultIsFailure(), mlirStringRefCreate(), mlir::python::DefaultingPyMlirContext::resolve(), and mlir::python::DefaultingPyLocation::resolve().

Referenced by bindDerived().

Member Data Documentation

◆ getInterfaceID

constexpr static GetTypeIDFunctionTy mlir::python::PyInferShapedTypeOpInterface::getInterfaceID

staticconstexpr

Initial value:

=

&mlirInferShapedTypeOpInterfaceTypeID

mlirInferShapedTypeOpInterfaceTypeID

MLIR_CAPI_EXPORTED MlirTypeID mlirInferShapedTypeOpInterfaceTypeID()

Returns the interface TypeID of the InferShapedTypeOpInterface.

Definition: Interfaces.cpp:123

Definition at line 406 of file IRInterfaces.cpp.

◆ pyClassName

constexpr static const char* mlir::python::PyInferShapedTypeOpInterface::pyClassName = "InferShapedTypeOpInterface"

staticconstexpr

Definition at line 405 of file IRInterfaces.cpp.

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

lib/Bindings/Python/IRInterfaces.cpp

Classes

Public Member Functions

Static Public Member Functions

Static Public Attributes

Additional Inherited Members