Searched defs:indices_format (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/sparse/ |
| H A D | _triton_ops.py | 892 indices_format, M, K, N, Ms, Ks, nbatches, SPLIT_N, compressed_sparse_tensor_as_key argument
988 bsr, other, indices_format="bsr_strided_mm_compressed", **meta_input argument
|