Home
last modified time | relevance | path

Searched refs:get_dimensions (Results 1 – 2 of 2) sorted by relevance

/llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/taco/
H A Dunit_test_tensor_core.py446 passed += (stats.get_dimensions(0) == (I,))
483 passed += (stats.get_dimensions(0) == (I, J))
485 passed += (stats.get_dimensions(1) == (I, J))
522 passed += (stats.get_dimensions(0) == (I, K))
524 passed += (stats.get_dimensions(1) == (I, K))
/llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/taco/tools/
H A Dmlir_pytaco.py939 def get_dimensions(self, idx: int) -> Tuple[int]: member in _Stats