Home
last modified time | relevance | path

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 DSparseCsrTensor.cpp158 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 DSparseBlasImpl.cpp37 …return AT_DISPATCH_PLAIN_SPARSE_COMPRESSED_LAYOUTS(compressed.layout(), "operands_support_triton_m… in operands_support_triton_mm_kernel()
H A DSparseCsrTensorMath.cpp359 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 DSparseCsrTensorImpl.cpp131 AT_DISPATCH_PLAIN_SPARSE_COMPRESSED_LAYOUTS(layout_, in resize_and_clear_()
H A DSparseCsrTensorUtils.h52 #define AT_DISPATCH_PLAIN_SPARSE_COMPRESSED_LAYOUTS( \ macro
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorConversions.cpp1878 AT_DISPATCH_PLAIN_SPARSE_COMPRESSED_LAYOUTS(layout, "sparse_compressed_to_sparse", in sparse_compressed_to_sparse()
H A DTensorShape.cpp3082 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()