Searched refs:input_values_ptr (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | sparse_slice_op_gpu.cu.cc | 164 const T* input_values_ptr = input_values.vec<T>().data(); in operator ()() local
207 input_indices_ptr, input_values, input_values_ptr, output_nnz_t, in operator ()()
242 input_indices_ptr, input_values_ptr, in operator ()()
|
| H A D | sparse_split_op_gpu.cu.cc | 223 const T* input_values_ptr = input_values.vec<T>().data(); in operator ()() local
306 input_values, input_values_ptr, sort_permutation, sort_permutation_ptr, in operator ()()
330 input_values_ptr, output_indices.data(), output_values.data()), in operator ()()
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | c_test_util.cc | 512 TF_Tensor* const* input_values_ptr = in Run() local
521 TF_SessionRun(session_, nullptr, inputs_ptr, input_values_ptr, inputs_.size(), in Run()
|