Searched refs:emit_compare_callback (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | sort_util.cc | 57 const EmitCallToNestedComputationCallback& emit_compare_callback, in EmitCompareLoopBody() argument
125 emit_compare_callback(values_to_compare, compare_return_buffer)); in EmitCompareLoopBody()
150 const EmitCallToNestedComputationCallback& emit_compare_callback, in EmitTiledCompareLoop() argument
249 element_address, write_element, emit_compare_callback, b); in EmitTiledCompareLoop()
255 emit_compare_callback, b, in EmitTiledCompareLoop()
262 emit_compare_callback, b, in EmitTiledCompareLoop()
299 const EmitCallToNestedComputationCallback& emit_compare_callback) { in EmitSortInPlace() argument
366 values_arrays, param_shmem_buffers, tile_size, emit_compare_callback, in EmitSortInPlace()
387 emit_compare_callback, b)); in EmitSortInPlace()
|
| D | sort_util.h | 42 const EmitCallToNestedComputationCallback& emit_compare_callback);
|