Searched refs:_DimInfo (Results 1 – 1 of 1) sorted by relevance
| /llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/taco/tools/ |
| H A D | mlir_pytaco.py | 1570 class _DimInfo: class
1581 def _get_dummy_dim_info() -> _DimInfo:
1583 return _DimInfo(-1, ModeFormat.DENSE)
1604 dim_infos: Tuple[_DimInfo, ...]
1819 index_to_dim_info: Dict[IndexVar, _DimInfo], argument
1821 dim_infos: Tuple[_DimInfo, ...], argument
1855 index_to_dim_info[i] = _DimInfo(d.dim, mode_format)
1898 dim_infos = tuple([_DimInfo(d, m) for d, m in zip(src_dims, mode_formats)])
|