Home
last modified time | relevance | path

Searched refs:EmitArrayElementAddress (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dir_array.cc441 llvm::Value* IrArray::EmitArrayElementAddress(const IrArray::Index& index, in EmitArrayElementAddress() function in xla::llvm_ir::IrArray
506 EmitArrayElementAddress(index, b, name, use_linear_index); in EmitReadArrayElement()
516 EmitArrayElementAddress(index, b, "", use_linear_index); in EmitWriteArrayElement()
Dir_array.h234 llvm::Value* EmitArrayElementAddress(const Index& index, llvm::IRBuilder<>* b,
Dsort_util.cc374 return values_arrays[operand].EmitArrayElementAddress(keys_index, b); in EmitSortInPlace()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc788 operand_array.EmitArrayElementAddress(operand_index, &b_); in HandleSelectAndScatter()
1579 arg_array.EmitArrayElementAddress(input_index, &b_), b_.getInt8PtrTy()); in EmitInnerLoopForVectorizedReduction()
1748 target_array.EmitArrayElementAddress(array_index, &b_); in EmitVectorizedReduce()
1782 target_array.EmitArrayElementAddress(array_index, &b_); in EmitVectorizedReduce()
1944 target_array.EmitArrayElementAddress(target_index, &b_, "slice.dest"); in HandleSlice()
1946 source_array.EmitArrayElementAddress(source_index, &b_, "slice.source"); in HandleSlice()
2500 target_array.EmitArrayElementAddress(target_index, &b_, "target_region"), in EmitFastConcatenate()
2518 source_array.EmitArrayElementAddress(source_index, &b_, "src_addr"), in EmitFastConcatenate()
Ddot_op_emitter.cc1117 llvm::Value* slice_ptr = outer_array.EmitArrayElementAddress(slice_index, b); in SliceOutInnerArray()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc2357 operand_array.EmitArrayElementAddress(operand_index, &b_); in EmitSelectAndScatterFromMlir()
2402 source_array.EmitArrayElementAddress(source_index, &b_); in EmitSelectAndScatterFromMlir()
2406 out_array.EmitArrayElementAddress(selected_index, &b_); in EmitSelectAndScatterFromMlir()
2478 llvm::Value* output_address = ir_arrays[0].EmitArrayElementAddress( in EmitRngGetAndUpdateState()
2700 desc.output.EmitArrayElementAddress(input_window_index, &b_); in EmitScatter()
4455 llvm::Value* output_address = output_array.EmitArrayElementAddress( in EmitEpilogueForReduction()