Searched refs:EncodingAttr (Results 1 – 6 of 6) sorted by relevance
| /llvm-project-15.0.7/mlir/test/python/dialects/sparse_tensor/ |
| H A D | dialect.py | 21 casted = st.EncodingAttr(parsed)
36 created = st.EncodingAttr.get(casted.dim_level_types, None, 16, 32)
58 casted = st.EncodingAttr(parsed)
71 created = st.EncodingAttr.get(casted.dim_level_types, casted.dim_ordering,
82 encoding = st.EncodingAttr(Attribute.parse(
|
| /llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/python/ |
| H A D | test_output.py | 20 def boilerplate(attr: st.EncodingAttr): argument
51 def build_compile_and_run_output(attr: st.EncodingAttr, compiler): argument
92 attr = st.EncodingAttr.get(level, ordering, bwidth, bwidth)
|
| H A D | test_SpMM.py | 29 def build_SpMM(attr: st.EncodingAttr): argument
51 def boilerplate(attr: st.EncodingAttr): argument
71 def build_compile_and_run_SpMM(attr: st.EncodingAttr, compiler): argument
145 attr = st.EncodingAttr.get(level, ordering, pwidth, iwidth)
|
| H A D | test_SDDMM.py | 31 def build_SDDMM(attr: st.EncodingAttr): argument
54 def boilerplate(attr: st.EncodingAttr): argument
71 def build_compile_and_run_SDDMMM(attr: st.EncodingAttr, compiler): argument
151 attr = st.EncodingAttr.get(level, ordering, pwidth, iwidth)
|
| H A D | test_stress.py | 221 attr = st.EncodingAttr.get(level, ordering, pwidth, iwidth)
|
| /llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/taco/tools/ |
| H A D | mlir_pytaco.py | 360 def mlir_tensor_attr(self) -> Optional[sparse_tensor.EncodingAttr]:
366 return sparse_tensor.EncodingAttr.get(mlir_storage_format,
853 attr: Optional[sparse_tensor.EncodingAttr]) -> ir.RankedTensorType: argument
|