Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dir_array.cc332 llvm::Value* IrArray::EmitArrayElementAddress(const IrArray::Index& index, in EmitArrayElementAddress() function in xla::llvm_ir::IrArray
396 EmitArrayElementAddress(index, b, name, use_linear_index); in EmitReadArrayElement()
406 EmitArrayElementAddress(index, b, "", use_linear_index); in EmitWriteArrayElement()
Dir_array.h227 llvm::Value* EmitArrayElementAddress(const Index& index, llvm::IRBuilder<>* b,
Dsort_util.cc372 return values_arrays[operand].EmitArrayElementAddress(keys_index, b); in EmitSortInPlace()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc869 operand_array.EmitArrayElementAddress(operand_index, &b_); in HandleSelectAndScatter()
1614 arg_array.EmitArrayElementAddress(input_index, &b_), b_.getInt8PtrTy()); in EmitInnerLoopForVectorizedReduction()
1768 target_array.EmitArrayElementAddress(array_index, &b_); in EmitVectorizedReduce()
1802 target_array.EmitArrayElementAddress(array_index, &b_); in EmitVectorizedReduce()
2030 target_array.EmitArrayElementAddress(target_index, &b_, "slice.dest"); in HandleSlice()
2032 source_array.EmitArrayElementAddress(source_index, &b_, "slice.source"); in HandleSlice()
2439 target_array.EmitArrayElementAddress(target_index, &b_, "target_region"), in EmitFastConcatenate()
2455 source_array.EmitArrayElementAddress(target_index, &b_, "src_addr"), in EmitFastConcatenate()
Ddot_op_emitter.cc921 llvm::Value* slice_ptr = outer_array.EmitArrayElementAddress(slice_index, b); in SliceOutInnerArray()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc713 .EmitArrayElementAddress(index, &b_, in EmitExtraOutputsForReduce()
913 operand_array.EmitArrayElementAddress(operand_index, &b_); in HandleSelectAndScatter()
951 .EmitArrayElementAddress(source_index, &b_); in HandleSelectAndScatter()
957 .EmitArrayElementAddress(selected_index, &b_); in HandleSelectAndScatter()
1181 .EmitArrayElementAddress(input_window_index, &b_); in EmitScatter()
2827 .EmitArrayElementAddress(element_index, &b_, in EmitEpilogueForReduction()
Dir_emitter.cc713 GetIrArray(*arg, *reduce).EmitArrayElementAddress(input_index, &b_); in HandleReduce()