mlir.dialects.transform.sparse_tensor

Classes

MatchSparseInOut	Checks if the payload op has any sparse inputs and/or outputs.
MatchSparseInOutAdaptor

Functions

sparse_tensor_match_sparse_inout(→ _ods_ir)

Module Contents

class mlir.dialects.transform.sparse_tensor.MatchSparseInOut(result, target, *, loc=None, ip=None)

Bases: _ods_ir

Checks if the payload op has any sparse inputs and/or outputs.

OPERATION_NAME = 'transform.sparse_tensor.match.sparse_inout'

_ODS_REGIONS = (0, True)

target() → _ods_ir

result() → _ods_ir

Shortcut to get an op result if it has only one (throws an error otherwise).

class mlir.dialects.transform.sparse_tensor.MatchSparseInOutAdaptor(operands: list, attributes: OpAttributeMap)

class mlir.dialects.transform.sparse_tensor.MatchSparseInOutAdaptor(operands: list, opview: OpView)

Bases: _ods_ir

OPERATION_NAME = 'transform.sparse_tensor.match.sparse_inout'

target() → _ods_ir

mlir.dialects.transform.sparse_tensor.sparse_tensor_match_sparse_inout(result, target, *, loc=None, ip=None) → _ods_ir