Searched refs:EmitWriteArrayElement (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | loop_emitter.cc | 44 target_array.EmitWriteArrayElement(array_index, target_element, b); in LoopEmitter()
63 target_arrays[i].EmitWriteArrayElement( in MakeBodyEmitterForMultiOutputFusion()
|
| D | sort_util.cc | 277 params[i].EmitWriteArrayElement(keys_index, value, b); in EmitTiledCompareLoop()
380 values_arrays[operand].EmitWriteArrayElement(keys_index, value, b); in EmitSortInPlace()
|
| D | ir_array.h | 251 void EmitWriteArrayElement(const Index& index, llvm::Value* value,
|
| D | dynamic_update_slice_util.cc | 94 output_array.EmitWriteArrayElement(output_index, update_data, b); in EmitDynamicUpdateSliceInPlaceImpl()
|
| D | ir_array.cc | 402 void IrArray::EmitWriteArrayElement(const Index& index, llvm::Value* value, in EmitWriteArrayElement() function in xla::llvm_ir::IrArray
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 501 target_array.EmitWriteArrayElement(/*index=*/element_index, result, &b_); in HandleDot()
625 target_array.EmitWriteArrayElement( in HandleDot()
|
| D | ir_emitter_unnested.cc | 2473 .EmitWriteArrayElement(index, load_from_shmem_buffer, &b_); in EmitTileElementForCopy()
2511 output_arrays[i].EmitWriteArrayElement( in EmitTileElementForFusion()
2515 output_arrays[0].EmitWriteArrayElement(untiled_index, output_value, &b_); in EmitTileElementForFusion()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 539 target_array_.EmitWriteArrayElement(target_index, result, b_); in EmitNaiveLlvmIrGemm()
577 target_array_.EmitWriteArrayElement(/*index=*/element_index, result, b_); in EmitScalarDot()
|
| D | ir_emitter.cc | 909 output_array.EmitWriteArrayElement(selected_index, scatter_value, &b_); in HandleSelectAndScatter()
2143 output_array.EmitWriteArrayElement(output_index, operand_data, &b_); in HandlePad()
|