Searched defs:mat1_meta (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/ |
| H A D | SparseSemiStructuredOps.cu | 524 const Tensor& mat1, const Tensor& mat1_meta, const Tensor& mat2, in sparse_semi_structured_mad_op()
789 const Tensor& mat1, const Tensor& mat1_meta, const Tensor& mat2, in _sparse_semi_structured_mm()
798 const Tensor& input, const Tensor& mat1, const Tensor& mat1_meta, in _sparse_semi_structured_addmm()
|
| /aosp_15_r20/external/pytorch/torch/_inductor/kernel/ |
| H A D | mm.py | 471 mat1, mat1_meta, mat2, *, out_dtype=None, layout=None argument
|