Searched defs:res_values (Results 1 – 7 of 7) sorted by relevance
| /external/pytorch/aten/src/ATen/native/sparse/ |
| D | SparseBinaryOpIntersectionKernel.cpp | 61 auto res_values = iter.tensor(0); in apply() local
94 accscalar_t res_values = 0; in apply() local
|
| D | SparseBinaryOpIntersectionCommon.h | 93 auto res_values = at::empty(res_values_sizes, lhs_values.options()); in make_value_selection_intersection_iter() local
401 const auto res_values = value_selection_intersection_kernel_t::apply( variable
|
| D | SparseTensor.cpp | 748 const auto res_values = values.has_value() ? *values : t._values(); in sparse_mask_like_prepare_sparse_inputs() local
763 const auto res_values = t._values().index_select(0, argsort_indices_hash); in sparse_mask_like_prepare_sparse_inputs() local
|
| D | SparseTensorMath.cpp | 885 const auto res_values = is_same_tensor(s_, res) ? values : values.to(res.scalar_type()); in intersection_binary_op_sparse_dense_out() local
958 const auto res_values = op(d_filtered, s_values).reshape(res_values_shape); in intersection_binary_op_sparse_dense_out() local
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | SparseBinaryOpIntersectionKernel.cu | 113 accscalar_t res_values = 0; in binary_op_intersection_kernel() local
146 auto res_values = iter.tensor(0); in apply() local
|
| D | Indexing.cu | 1653 const auto res_values = values.index_select(0, selected_dim_indices); in index_select_sparse_cuda() local
1797 const auto res_values = values.index_select(dim - sparse_dim + 1, index); in index_select_sparse_cuda() local
|
| /external/pytorch/aten/src/ATen/native/ |
| D | TensorShape.cpp | 1942 const auto res_values = index_select(values, 0, index); in index_select_sparse_cpu() local
2415 const auto res_values = index_select(values, 0, selected_dim_indices); in index_select_sparse_cpu() local
2492 const auto res_values = index_select(values, dim - sparse_dim + 1, index); in index_select_sparse_cpu() local
|