Searched refs:EmitArrayElementAddress (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| H A D | ir_array.cc | 470 llvm::Value* IrArray::EmitArrayElementAddress(const IrArray::Index& index, in EmitArrayElementAddress() function in xla::llvm_ir::IrArray
536 EmitArrayElementAddress(index, b, name, use_linear_index); in EmitReadArrayElement()
547 EmitArrayElementAddress(index, b, "", use_linear_index); in EmitWriteArrayElement()
|
| H A D | ir_array.h | 245 llvm::Value* EmitArrayElementAddress(const Index& index, llvm::IRBuilder<>* b,
|
| H A D | sort_util.cc | 399 return values_arrays[operand].EmitArrayElementAddress(keys_index, b); in EmitSortInPlace()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | ir_emitter.cc | 815 operand_array.EmitArrayElementAddress(operand_index, &b_); in HandleSelectAndScatter()
1677 arg_array.EmitArrayElementAddress(input_index, &b_), b_.getInt8PtrTy()); in EmitInnerLoopForVectorizedReduction()
1850 target_array.EmitArrayElementAddress(array_index, &b_); in EmitVectorizedReduce()
1888 target_array.EmitArrayElementAddress(array_index, &b_); in EmitVectorizedReduce()
2055 target_array.EmitArrayElementAddress(target_index, &b_, "slice.dest"); in HandleSlice()
2057 source_array.EmitArrayElementAddress(source_index, &b_, "slice.source"); in HandleSlice()
2661 target_array.EmitArrayElementAddress(target_index, &b_, "target_region"), in EmitFastConcatenate()
2679 source_array.EmitArrayElementAddress(source_index, &b_, "src_addr"), in EmitFastConcatenate()
|
| H A D | dot_op_emitter.cc | 1315 llvm::Value* slice_ptr = outer_array.EmitArrayElementAddress(slice_index, b); in SliceOutInnerArray()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | ir_emitter_unnested.cc | 2072 operand_array.EmitArrayElementAddress(operand_index, &b_); in EmitSelectAndScatter()
2128 source_array.EmitArrayElementAddress(source_index, &b_); in EmitSelectAndScatter()
2132 out_array.EmitArrayElementAddress(selected_index, &b_); in EmitSelectAndScatter()
2191 llvm::Value* output_address = ir_arrays[0].EmitArrayElementAddress( in EmitRngGetAndUpdateState()
2413 desc.output.EmitArrayElementAddress(input_window_index, &b_); in EmitScatter()
3782 return output_array.EmitArrayElementAddress(output_index, &b_, in GetOutputAddressForReduction()
|