Home
last modified time | relevance | path

Searched defs:IndexedHeap (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dtopk_op_gpu.h106 struct IndexedHeap { struct
109 __device__ IndexedHeap(const Data<T>& d) : data(d) {} in IndexedHeap() argument
111 __device__ bool is_above(int left, int right) { in is_above()
128 __device__ void assign(int i, const Entry& entry) { data[i] = entry; } in assign()
130 __device__ void push_up(int i) { in push_up()
143 __device__ void swap(int a, int b) { in swap()
149 __device__ void push_root_down(int k) { push_down(0, k); } in push_root_down()
152 __device__ void push_down(int node, int k) { in push_down()
172 __device__ void build(int k) { in build()
179 __device__ void remove_root(int k) { in remove_root()
[all …]