Home
last modified time | relevance | path

Searched refs:sorted_input_unique_ids_ptr (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dunique_op_gpu.cu.h301 TIndex* sorted_input_unique_ids_ptr = nullptr; in ComputeAsync() local
303 &sorted_input_unique_ids_ptr, done); in ComputeAsync()
309 sorted_input_unique_ids_ptr), in ComputeAsync()
320 const_cast<TIndex*>(sorted_input_unique_ids_ptr) + in ComputeAsync()
330 sorted_input_unique_ids_ptr, last_idx_host, in ComputeAsync()
356 sorted_input_unique_ids_ptr, unique_input_inds_ptr, in ComputeAsync()
436 sorted_input_unique_ids_ptr, in ComputeAsync()