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()
|