Home
last modified time | relevance | path

Searched refs:ModeFormat (Results 1 – 5 of 5) sorted by relevance

/llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/taco/tools/
H A Dmlir_pytaco_api.py28 mode_format = mlir_pytaco.ModeFormat
50 compressed = mlir_pytaco.ModeFormat.COMPRESSED
51 Compressed = mlir_pytaco.ModeFormat.COMPRESSED
52 dense = mlir_pytaco.ModeFormat.DENSE
53 Dense = mlir_pytaco.ModeFormat.DENSE
H A Dmlir_pytaco.py181 class ModeFormat(enum.Enum): class
193 def _mode_format_operation(a: ModeFormat, b: ModeFormat, argument
196 return (ModeFormat.COMPRESSED
197 if op(a == ModeFormat.COMPRESSED, b == ModeFormat.COMPRESSED) else
198 ModeFormat.DENSE)
284 formats: List[ModeFormat]
990 def _init_format(self, fmt: Union[ModeFormat, List[ModeFormat], argument
1006 if isinstance(fmt, ModeFormat):
1026 fmt: Optional[Union[ModeFormat, List[ModeFormat], argument
1071 fmt = fmt or ModeFormat.COMPRESSED
[all …]
/llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/taco/
H A Dunit_test_tensor_utils.py18 _COMPRESSED = mlir_pytaco.ModeFormat.COMPRESSED
19 _DENSE = mlir_pytaco.ModeFormat.DENSE
H A Dunit_test_tensor_io.py19 _COMPRESSED = mlir_pytaco.ModeFormat.COMPRESSED
20 _DENSE = mlir_pytaco.ModeFormat.DENSE
H A Dunit_test_tensor_core.py16 _COMPRESSED = mlir_pytaco.ModeFormat.COMPRESSED
17 _DENSE = mlir_pytaco.ModeFormat.DENSE