Searched refs:sorted_input_inds_ptr (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | unique_op_gpu.cu.h | 273 TIndex* sorted_input_inds_ptr = nullptr; in ComputeAsync() local
275 &sorted_input_inds_ptr, done); in ComputeAsync()
289 /*indices_out=*/sorted_input_inds_ptr), in ComputeAsync()
328 sorted_input_inds, sorted_input_inds_ptr, in ComputeAsync()
355 device, input_size, uniq_size, sorted_input_inds_ptr, in ComputeAsync()
435 LookupAndScatterUniqueIds(device, input_size, sorted_input_inds_ptr, in ComputeAsync()
|