Home
last modified time | relevance | path

Searched defs:r_values (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/native/sparse/
DSparseTensorMath.cpp125 Tensor r_values = r._values(); // Sigh... needed because mul_out takes Tensor& in mul_out_sparse_zerodim() local
179 Tensor r_values = r._values(); // Sigh... needed because pow_out takes Tensor& in pow_out_sparse_scalar() local
248 Tensor r_values = r._values(); // Sigh... needed because div_out takes Tensor& in div_out_sparse_zerodim() local
450 Tensor r_values = new_values_with_size_of(s_values, max_nnz).zero_(); in add_out_sparse_contiguous() local
541 Tensor r_values = at::cat({t_values, s_values}, 0).to(r.scalar_type()); in add_out_sparse_non_contiguous() local
1173 Tensor r_values = r_buffer.to(r.scalar_type()); in mul_out_sparse_cpu() local