Searched refs:AT_DISPATCH_PLAIN_SPARSE_COMPRESSED_LAYOUTS (Results 1 – 7 of 7) sorted by relevance
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
H A D | SparseCsrTensor.cpp | 158 const int block_ndim = AT_DISPATCH_PLAIN_SPARSE_COMPRESSED_LAYOUTS( in _validate_sparse_compressed_tensor_args_worker() 525 …const int block_ndim = AT_DISPATCH_PLAIN_SPARSE_COMPRESSED_LAYOUTS(layout, "estimate_sparse_compre… in _estimate_sparse_compressed_tensor_size() 779 AT_DISPATCH_PLAIN_SPARSE_COMPRESSED_LAYOUTS(self.layout(), "copy_sparse_compressed_", in copy_sparse_compressed_() 1036 AT_DISPATCH_PLAIN_SPARSE_COMPRESSED_LAYOUTS(self.layout(), "select", [&] {}, [&] { in select_sparse_csr_worker() 1076 AT_DISPATCH_PLAIN_SPARSE_COMPRESSED_LAYOUTS(self.layout(), "select", [&]() {}, in select_sparse_csr_worker() 1233 Tensor new_values = AT_DISPATCH_PLAIN_SPARSE_COMPRESSED_LAYOUTS( in select_sparse_csr_worker()
|
H A D | SparseBlasImpl.cpp | 37 …return AT_DISPATCH_PLAIN_SPARSE_COMPRESSED_LAYOUTS(compressed.layout(), "operands_support_triton_m… in operands_support_triton_mm_kernel()
|
H A D | SparseCsrTensorMath.cpp | 359 return AT_DISPATCH_PLAIN_SPARSE_COMPRESSED_LAYOUTS( in sparse_mask_sparse_compressed() 376 return AT_DISPATCH_PLAIN_SPARSE_COMPRESSED_LAYOUTS( in sparse_mask_sparse_compressed()
|
/aosp_15_r20/external/pytorch/aten/src/ATen/ |
H A D | SparseCsrTensorImpl.cpp | 131 AT_DISPATCH_PLAIN_SPARSE_COMPRESSED_LAYOUTS(layout_, in resize_and_clear_()
|
H A D | SparseCsrTensorUtils.h | 52 #define AT_DISPATCH_PLAIN_SPARSE_COMPRESSED_LAYOUTS( \ macro
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
H A D | TensorConversions.cpp | 1878 AT_DISPATCH_PLAIN_SPARSE_COMPRESSED_LAYOUTS(layout, "sparse_compressed_to_sparse", in sparse_compressed_to_sparse()
|
H A D | TensorShape.cpp | 3082 result_vals = AT_DISPATCH_PLAIN_SPARSE_COMPRESSED_LAYOUTS( in sparse_compressed_transpose() 3094 result_vals = AT_DISPATCH_PLAIN_SPARSE_COMPRESSED_LAYOUTS( in sparse_compressed_transpose()
|