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.cc96 param_x_array.EmitReadArrayElement(zero_2d, &b) in TEST_F()
107 concat1_array.EmitReadArrayElement(zero_2d, &b) in TEST_F()
118 concat2_array.EmitReadArrayElement(zero_2d, &b) in TEST_F()
128 add_array.EmitReadArrayElement(zero_2d, &b)->setName("read_add_array"); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Ddynamic_update_slice_util.cc174 return operand_arrays[2 + index].EmitReadArrayElement( in EmitDynamicUpdateSliceInPlace()
178 return update_array.EmitReadArrayElement(index, b); in EmitDynamicUpdateSliceInPlace()
Dfused_ir_emitter.cc105 .EmitReadArrayElement(index, b_, constant->name()); in HandleConstant()
Dir_array.h251 llvm::Value* EmitReadArrayElement(const Index& index, llvm::IRBuilder<>* b,
Dsort_util.cc197 auto value = params[i].EmitReadArrayElement(keys_index, b); in EmitTiledCompareLoop()
Dir_array.cc501 llvm::Value* IrArray::EmitReadArrayElement(const Index& index, in EmitReadArrayElement() function in xla::llvm_ir::IrArray
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc633 llvm::Value* lhs_element = lhs_array_.EmitReadArrayElement(lhs_index, b_); in EmitNaiveLlvmIrGemm()
634 llvm::Value* rhs_element = rhs_array_.EmitReadArrayElement(rhs_index, b_); in EmitNaiveLlvmIrGemm()
702 lhs_array_.EmitReadArrayElement(/*index=*/element_index, b_); in EmitScalarDot()
704 rhs_array_.EmitReadArrayElement(/*index=*/element_index, b_); in EmitScalarDot()
Dir_emitter.cc779 operand_array.EmitReadArrayElement(operand_index, &b_); in HandleSelectAndScatter()
819 source_array.EmitReadArrayElement(source_index, &b_); in HandleSelectAndScatter()
824 output_array.EmitReadArrayElement(selected_index, &b_); in HandleSelectAndScatter()
2038 operand_array.EmitReadArrayElement(operand_index, &b_); in HandlePad()
2175 GetIrArrayFor(hlo->operand(0)).EmitReadArrayElement(array_index, &b_); in HandleSliceToDynamic()
2243 GetIrArrayFor(hlo->operand(0)).EmitReadArrayElement(source_index, &b_); in HandlePadToStatic()
3215 return GetIrArrayFor(operand).EmitReadArrayElement(index, &b_); in DefaultAction()
3329 return GetIrArrayFor(operand).EmitReadArrayElement(index, &b_); in BindFusionArguments()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc95 .EmitReadArrayElement(index, &b_, operand->name()); in DefaultAction()
693 .EmitReadArrayElement(index, &b_, operand->name()); in BindFusionArguments()
Dir_emitter_unnested.cc919 source_array.EmitReadArrayElement(array_index, &b_, /*name=*/""), &b_, in EmitPadToStaticFromMlir()
1041 ir_arrays[0].EmitReadArrayElement(dyn_index, &b_, /*name=*/"", in EmitSliceToDynamicFromMlir()
1804 return ir_array.EmitReadArrayElement(index, builder); in EmitLoopFusionFromMlir()
1907 return ir_arrays[i].EmitReadArrayElement( in HandleFusion()
1951 return ir_arrays[i].EmitReadArrayElement( in HandleFusion()
2060 return ir_arrays[i].EmitReadArrayElement(index, &b_, in HandleFusion()
2347 operand_array.EmitReadArrayElement(operand_index, &b_); in EmitSelectAndScatterFromMlir()
2543 return scatter_indices.EmitReadArrayElement(index, &b_, in EmitScatterFromMlir()
2548 return updates.EmitReadArrayElement(index, &b_, "update"); in EmitScatterFromMlir()
3681 return init_array.EmitReadArrayElement(index, &b_); in BuildInitializerThunkForMlir()
[all …]