Searched defs:sorted_inputs (Results 1 – 6 of 6) 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
1090 auto sorted_inputs = sortReverseTopological(fused_cat->inputs()); in fuseConcats() local
2455 sorted_inputs, sorted_outputs, graph argument
2447 sorted_inputs, sorted_outputs, graph argument