Searched refs:comparison_values (Results 1 – 2 of 2) sorted by relevance
58 std::unique_ptr<char*[]> comparison_values(new char*[2 * values_count]); in __xla_cpu_runtime_KeyValueSort() local78 comparison_values[i * 2] = values[i] + memory_index_lhs; in __xla_cpu_runtime_KeyValueSort()79 comparison_values[i * 2 + 1] = values[i] + memory_index_rhs; in __xla_cpu_runtime_KeyValueSort()82 less_than(&result, run_options, comparison_values.get(), nullptr, in __xla_cpu_runtime_KeyValueSort()
302 comparison_values = last_prediction_chunk.shape[0]304 model_predictions[prediction_name][0, -comparison_values:])