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: _ods_ir, target: _ods_ir, *, loc: _ods_ir | None = None, ip: _ods_ir | None = 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[Value], attributes: OpAttributeMap)¶

class mlir.dialects.transform.sparse_tensor.MatchSparseInOutAdaptor(operands: list[Value], 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: _ods_ir, target: _ods_ir, *, loc: _ods_ir | None = None, ip: _ods_ir | None = None) → _ods_ir¶