Home
last modified time | relevance | path

Searched refs:AT_DISPATCH_PLAIN_SPARSE_COMPRESSED_LAYOUTS (Results 1 – 7 of 7) sorted by relevance

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