Searched defs:reduce_sparse_csr_cpu_template (Results 1 – 1 of 1) sorted by relevance
1278 Tensor reduce_sparse_csr_cpu_template(const Tensor& sparse, std::vector<int64_t> dims, ReductionOp … in reduce_sparse_csr_cpu_template() function1296 Tensor reduce_sparse_csr_cpu_template(const Tensor& sparse, IntArrayRef dims_to_sum, bool keepdim, … in reduce_sparse_csr_cpu_template() function