Home
last modified time | relevance | path

Searched refs:EmitReadArrayElement (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_noalias_test.cc94 param_x_array.EmitReadArrayElement(zero2D, &b) in TEST_F()
104 concat1_array.EmitReadArrayElement(zero2D, &b) in TEST_F()
114 concat2_array.EmitReadArrayElement(zero2D, &b) in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dfused_ir_emitter.cc98 .EmitReadArrayElement(index, b_); in HandleConstant()
136 .EmitReadArrayElement(index, b_); in HandleGetTupleElement()
164 .EmitReadArrayElement(index, b_); in HandleParameter()
Ddynamic_update_slice_util.cc120 return operand_arrays[2 + index].EmitReadArrayElement( in EmitDynamicUpdateSliceInPlace()
124 return update_array.EmitReadArrayElement(index, b); in EmitDynamicUpdateSliceInPlace()
Dir_array.h244 llvm::Value* EmitReadArrayElement(const Index& index, llvm::IRBuilder<>* b,
Dsort_util.cc196 auto value = params[i].EmitReadArrayElement(keys_index, b); in EmitTiledCompareLoop()
Dir_array.cc391 llvm::Value* IrArray::EmitReadArrayElement(const Index& index, in EmitReadArrayElement() function in xla::llvm_ir::IrArray
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc72 return GetIrArray(*operand, *hlo).EmitReadArrayElement(index, &b_); in DefaultAction()
486 lhs_array.EmitReadArrayElement(/*index=*/element_index, &b_); in HandleDot()
488 rhs_array.EmitReadArrayElement(/*index=*/element_index, &b_); in HandleDot()
581 llvm::Value* lhs_element = lhs_array.EmitReadArrayElement(lhs_index, &b_); in HandleDot()
584 llvm::Value* rhs_element = rhs_array.EmitReadArrayElement(rhs_index, &b_); in HandleDot()
Dir_emitter_unnested.cc903 operand_array.EmitReadArrayElement(operand_index, &b_); in HandleSelectAndScatter()
1003 return GetIrArray(*operand, *rng).EmitReadArrayElement(index, &b_); in HandleRng()
1055 .EmitReadArrayElement(index, &b_, "scatter_index"); in HandleScatter()
1060 .EmitReadArrayElement(index, &b_, "update"); in HandleScatter()
2016 .EmitReadArrayElement(index, &b_); in BuildInitializerThunk()
2689 .EmitReadArrayElement(IrArray::Index(b_.getInt32Ty()), &b_); in EmitPrologueForOneReduction()
2911 .EmitReadArrayElement(index, &b_); in EmitTileElementForReduction()
3162 Store(input_in_logical_shape.EmitReadArrayElement( in EmitKernel()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc490 llvm::Value* lhs_element = lhs_array_.EmitReadArrayElement(lhs_index, b_); in EmitNaiveLlvmIrGemm()
491 llvm::Value* rhs_element = rhs_array_.EmitReadArrayElement(rhs_index, b_); in EmitNaiveLlvmIrGemm()
553 lhs_array_.EmitReadArrayElement(/*index=*/element_index, b_); in EmitScalarDot()
555 rhs_array_.EmitReadArrayElement(/*index=*/element_index, b_); in EmitScalarDot()
Dir_emitter.cc860 operand_array.EmitReadArrayElement(operand_index, &b_); in HandleSelectAndScatter()
900 source_array.EmitReadArrayElement(source_index, &b_); in HandleSelectAndScatter()
905 output_array.EmitReadArrayElement(selected_index, &b_); in HandleSelectAndScatter()
2124 operand_array.EmitReadArrayElement(operand_index, &b_); in HandlePad()
2656 return GetIrArrayFor(operand).EmitReadArrayElement(index, &b_); in HandleRng()
3031 return GetIrArrayFor(operand).EmitReadArrayElement(index, &b_); in DefaultAction()