Searched refs:keys_index (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | sort_util.cc | 195 IrArray::Index keys_index(keys_multi_index, params[i].GetShape(), in EmitTiledCompareLoop() local
197 auto value = params[i].EmitReadArrayElement(keys_index, b); in EmitTiledCompareLoop()
274 IrArray::Index keys_index(keys_multi_index, params[i].GetShape(), in EmitTiledCompareLoop() local
279 params[i].EmitWriteArrayElement(keys_index, value, b); in EmitTiledCompareLoop()
362 IrArray::Index keys_index(keys_multi_index, values_arrays[0].GetShape(), in EmitSortInPlace() local
365 keys_index, dimension_to_sort, dimension_to_sort_bound, xor_masks, in EmitSortInPlace()
371 IrArray::Index keys_index(keys_multi_index, in EmitSortInPlace() local
374 return values_arrays[operand].EmitArrayElementAddress(keys_index, b); in EmitSortInPlace()
379 IrArray::Index keys_index(keys_multi_index, in EmitSortInPlace() local
382 values_arrays[operand].EmitWriteArrayElement(keys_index, value, b); in EmitSortInPlace()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.h | 207 const llvm_ir::IrArray::Index& keys_index,
|