Searched refs:EmitWriteArrayElement (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| H A D | loop_emitter.cc | 79 target_arrays_vec[0].EmitWriteArrayElement(array_index, target_element, in MakeBodyEmitter()
95 target_arrays_vec[i].EmitWriteArrayElement( in MakeBodyEmitter()
|
| H A D | sort_util.cc | 303 params[i].EmitWriteArrayElement(keys_index, value, b); in EmitTiledCompareLoop()
410 values_arrays[operand].EmitWriteArrayElement(keys_index, value, b); in EmitSortInPlace()
|
| H A D | dynamic_update_slice_util.cc | 148 output_array.EmitWriteArrayElement(output_index, update_data, b); in EmitDynamicUpdateSliceInPlaceImpl()
|
| H A D | ir_array.h | 269 void EmitWriteArrayElement(const Index& index, llvm::Value* value,
|
| H A D | ir_array.cc | 543 void IrArray::EmitWriteArrayElement(const Index& index, llvm::Value* value, in EmitWriteArrayElement() function in xla::llvm_ir::IrArray
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | dot_op_emitter.cc | 752 target_array_.EmitWriteArrayElement(target_index, result, b_); in EmitNaiveLlvmIrGemm()
790 target_array_.EmitWriteArrayElement(/*index=*/element_index, result, b_); in EmitScalarDot()
|
| H A D | ir_emitter.cc | 863 output_array.EmitWriteArrayElement(selected_index, scatter_value, &b_); in HandleSelectAndScatter()
2168 output_array.EmitWriteArrayElement(output_index, operand_data, &b_); in HandlePad()
2308 data_array.EmitWriteArrayElement(dest_index, source_element, &b_); in HandleSliceToDynamic()
2375 data_array.EmitWriteArrayElement(array_index, source_element, &b_); in HandlePadToStatic()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | ir_emitter_unnested.cc | 835 output_array.EmitWriteArrayElement( in EmitPadToStatic()
958 data_array.EmitWriteArrayElement( in EmitSliceToDynamic()
1881 result_ir[0].EmitWriteArrayElement( in EmitExtraOutputsForReduce()
3565 output_arrays[i].EmitWriteArrayElement( in EmitTileElementForTranspose()
3569 output_arrays[0].EmitWriteArrayElement(untiled_index, output_value, &b_); in EmitTileElementForTranspose()
5101 src_ir_array.EmitWriteArrayElement(slice_dst_index, input_ir_values[i], in EmitElementForInputFusibleSlices()
|