Searched refs:AT_DISPATCH_ALL_SPARSE_COMPRESSED_LAYOUTS (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | SparseCsrTensor.cpp | 130 …AT_DISPATCH_ALL_SPARSE_COMPRESSED_LAYOUTS(layout, "validate_sparse_compressed_tensor_args", [&]{}); in _validate_sparse_compressed_tensor_args_worker()
353 …Layout layout = AT_DISPATCH_ALL_SPARSE_COMPRESSED_LAYOUTS(options.layout(), "new_compressed_tensor… in new_compressed_tensor()
398 AT_DISPATCH_ALL_SPARSE_COMPRESSED_LAYOUTS(layout_, "sparse_compressed_tensor_with_dims", [&]{}); in sparse_compressed_tensor_with_dims()
465 AT_DISPATCH_ALL_SPARSE_COMPRESSED_LAYOUTS(layout_, "sparse_compressed_tensor_unsafe", [&]{}); in _sparse_compressed_tensor_unsafe_symint()
593 AT_DISPATCH_ALL_SPARSE_COMPRESSED_LAYOUTS(layout_, "sparse_compressed_tensor", [&]{}); in sparse_compressed_tensor()
622 AT_DISPATCH_ALL_SPARSE_COMPRESSED_LAYOUTS(layout_, "sparse_compressed_tensor", [&]{}); in sparse_compressed_tensor()
754 AT_DISPATCH_ALL_SPARSE_COMPRESSED_LAYOUTS(self.layout(), "copy_sparse_compressed_", [&]{}); in copy_sparse_compressed_()
865 AT_DISPATCH_ALL_SPARSE_COMPRESSED_LAYOUTS( in resize_as_sparse_compressed_()
867 AT_DISPATCH_ALL_SPARSE_COMPRESSED_LAYOUTS( in resize_as_sparse_compressed_()
971 AT_DISPATCH_ALL_SPARSE_COMPRESSED_LAYOUTS( in select_sparse_csr_worker()
|
| H A D | SparseCsrTensorMath.cpp | 188 AT_DISPATCH_ALL_SPARSE_COMPRESSED_LAYOUTS(self.layout(), "unary_op_inplace", [](){}); in unary_op_inplace()
412 AT_DISPATCH_ALL_SPARSE_COMPRESSED_LAYOUTS(self.layout(), "zero_sparse_csr_", [](){}); in zero_sparse_csr_()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | SparseCsrTensorUtils.h | 16 #define AT_DISPATCH_ALL_SPARSE_COMPRESSED_LAYOUTS(LAYOUT, NAME, ...) \ macro
176 AT_DISPATCH_ALL_SPARSE_COMPRESSED_LAYOUTS( in get_sparse_csr_impl()
|