Searched refs:sparse_sampled_addmm_check_inputs (Results 1 – 3 of 3) sorted by relevance
10 TORCH_API void sparse_sampled_addmm_check_inputs(
124 at::native::sparse::sparse_sampled_addmm_check_inputs(self, mat1, mat2, beta, alpha, result); in sparse_sampled_addmm_out_sparse_csr_cpu()166 void sparse_sampled_addmm_check_inputs( in sparse_sampled_addmm_check_inputs() function
45 at::native::sparse::sparse_sampled_addmm_check_inputs( in sparse_sampled_addmm_out_sparse_csr_cuda()