mlir.dialects._sparse_tensor_transform_ops_gen

Attributes

Classes

MatchSparseInOut

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

Functions

Module Contents

mlir.dialects._sparse_tensor_transform_ops_gen._ods_ir
class mlir.dialects._sparse_tensor_transform_ops_gen.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).

mlir.dialects._sparse_tensor_transform_ops_gen.sparse_tensor_match_sparse_inout(result, target, *, loc=None, ip=None) _ods_ir