| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | SegmentReduce.cu | 90 const scalar_t* values_data, in segment_reduce_forward_kernel()
162 const scalar_t* values_data, in segment_reduce_backward_kernel()
326 const auto* values_data = data_contig.const_data_ptr<scalar_t>(); in _segment_reduce_lengths_offsets_backward_cuda_kernel() local
|
| H A D | TensorModeKernel.cu | 162 scalar_t* values_data = values.mutable_data_ptr<scalar_t>(); in calculate_mode() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | SegmentReduce.cpp | 55 const auto* values_data = data.const_data_ptr<scalar_t>(); in _segment_reduce_lengths_cpu_kernel1() local
217 const auto* values_data = data_contig.const_data_ptr<scalar_t>(); in _segment_reduce_cpu_lengths_backward_kernel1() local
|
| H A D | TensorDimApply.h | 14 T1* values_data = values.data_ptr<T1>(); in tensor_dim_apply3() local
|
| H A D | ReduceOps.cpp | 772 void cummax_cummin_helper(const T1* self_data, T1* values_data, T2* indices_data, in cummax_cummin_helper()
|
| H A D | BatchLinearAlgebra.cpp | 2804 auto values_data = complex_values.const_data_ptr<c10::complex<scalar_t>>(); in linalg_eig_make_complex_eigenvectors_impl() local
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/ |
| H A D | op_index_put.cpp | 124 const CTYPE* const values_data = values.const_data_ptr<CTYPE>(); in index_put_out() local
|
| H A D | op_topk.cpp | 64 CTYPE* values_data = values.mutable_data_ptr<CTYPE>(); in perform_topk() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | SpmmReduceKernel.cpp | 360 auto values_data = values.accessor<scalar_t, 1>(); in spmm_reduce_normalize_values_kernel_impl() local
394 auto values_data = values.accessor<const scalar_t, 1>(); in spmm_reduce_backward_other_arg_kernel_impl() local
|
| H A D | TensorCompareKernel.cpp | 257 scalar_t* values_data = (scalar_t*)values_data_bytes; in mode_kernel_impl() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/linalg/ |
| H A D | BatchLinearAlgebraLib.cpp | 1229 auto values_data = values.data_ptr<value_t>(); in apply_syevd() local
1316 auto values_data = values.data_ptr<value_t>(); in apply_syevj() local
1370 auto values_data = values.data_ptr<value_t>(); in apply_syevj_batched() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | lookup_table_op.h | 277 auto values_data = values->flat<V>(); in ExportValues() local
|
| H A D | lookup_table_op.cc | 203 auto values_data = values->flat<V>(); in ExportKeysAndValues() local
393 auto values_data = values->matrix<V>(); in ExportKeysAndValues() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ctc/ |
| H A D | ctc_beam_search_decoder.cc | 158 int32_t* values_data = GetTensorData<int32_t>(values); in StoreAllDecodedSequences() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/examples/custom_ops_doc/simple_hash_table/ |
| H A D | simple_hash_table_kernel.cc | 82 auto values_data = values->flat<V>(); in Export() local
|