Searched refs:EmitWriteArrayElement (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | loop_emitter.cc | 52 target_array.EmitWriteArrayElement(array_index, target_element, b); in LoopEmitter()
71 target_arrays[i].EmitWriteArrayElement( in MakeBodyEmitterForMultiOutput()
|
| D | sort_util.cc | 279 params[i].EmitWriteArrayElement(keys_index, value, b); in EmitTiledCompareLoop()
382 values_arrays[operand].EmitWriteArrayElement(keys_index, value, b); in EmitSortInPlace()
|
| D | ir_array.h | 258 void EmitWriteArrayElement(const Index& index, llvm::Value* value,
|
| D | dynamic_update_slice_util.cc | 148 output_array.EmitWriteArrayElement(output_index, update_data, b); in EmitDynamicUpdateSliceInPlaceImpl()
|
| D | ir_array.cc | 512 void IrArray::EmitWriteArrayElement(const Index& index, llvm::Value* value, in EmitWriteArrayElement() function in xla::llvm_ir::IrArray
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 688 target_array_.EmitWriteArrayElement(target_index, result, b_); in EmitNaiveLlvmIrGemm()
726 target_array_.EmitWriteArrayElement(/*index=*/element_index, result, b_); in EmitScalarDot()
|
| D | ir_emitter.cc | 828 output_array.EmitWriteArrayElement(selected_index, scatter_value, &b_); in HandleSelectAndScatter()
2057 output_array.EmitWriteArrayElement(output_index, operand_data, &b_); in HandlePad()
2180 data_array.EmitWriteArrayElement(dest_index, source_element, &b_); in HandleSliceToDynamic()
2244 data_array.EmitWriteArrayElement(array_index, source_element, &b_); in HandlePadToStatic()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 917 output_array.EmitWriteArrayElement( in EmitPadToStaticFromMlir()
1039 data_array.EmitWriteArrayElement( in EmitSliceToDynamicFromMlir()
2131 result_ir_arrays[extra_output_gens[i].second].EmitWriteArrayElement( in EmitExtraOutputsForReduce()
4110 .EmitWriteArrayElement(index, load_from_shmem_buffer, &b_); in EmitTileElementForCopy()
4192 output_arrays[i].EmitWriteArrayElement( in EmitTileElementForFusion()
4196 output_arrays[0].EmitWriteArrayElement(untiled_index, output_value, &b_); in EmitTileElementForFusion()
5786 src_ir_array.EmitWriteArrayElement(slice_dst_index, input_ir_values[i], in EmitElementForInputFusibleSlices()
|