Searched defs:sorted_inputs (Results 1 – 4 of 4) 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()
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
37 const xla::XlaOp sorted_inputs = ctx->Input("sorted_inputs"); in BuildLowerUpperBoundOp() local
308 llvm::SmallVectorImpl<Operation*>* sorted_inputs) { in SortTPUReplicatedInputsByIndex()