Searched defs:sorted_inputs (Results 1 – 2 of 2) sorted by relevance
35 __global__ void UpperBoundKernel(const T* sorted_inputs, int batch_size, in UpperBoundKernel()47 __global__ void LowerBoundKernel(const T* sorted_inputs, int batch_size, in LowerBoundKernel()
112 const auto sorted_inputs = sorted_inputs_t.template flat<T>(); in Compute() local153 const auto sorted_inputs = sorted_inputs_t.template flat<T>(); in Compute() local