Searched defs:sorted_inputs (Results 1 – 3 of 3) sorted by relevance
34 __global__ void UpperBoundKernel(const T* __restrict__ sorted_inputs, in UpperBoundKernel()47 __global__ void LowerBoundKernel(const T* __restrict__ sorted_inputs, in LowerBoundKernel()
131 const auto sorted_inputs = sorted_inputs_t.template flat<T>(); in Compute() local184 const auto sorted_inputs = sorted_inputs_t.template flat<T>(); in Compute() local
37 const xla::XlaOp sorted_inputs = ctx->Input("sorted_inputs"); in BuildLowerUpperBoundOp() local