Home
last modified time | relevance | path

Searched refs:GetIndVarValue (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dparallel_loop_emitter.cc63 array_multi_index[dimension] = loop->GetIndVarValue(); in EmitIndexAndSetExitBasicBlock()
70 array_multi_index[dimension] = loop->GetIndVarValue(); in EmitIndexAndSetExitBasicBlock()
Ddot_op_emitter.cc600 lhs_multi_index[lhs_reduction_dimension] = reduction_loop->GetIndVarValue(); in EmitNaiveLlvmIrGemm()
603 rhs_multi_index[rhs_reduction_dimension] = reduction_loop->GetIndVarValue(); in EmitNaiveLlvmIrGemm()
Dir_emitter.cc1712 array_multi_index[dimension] = loop->GetIndVarValue(); in EmitVectorizedReduce()
1733 array_multi_index[innermost_dimension] = loop->GetIndVarValue(); in EmitVectorizedReduce()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dkernel_support_library.cc51 for_body_generator(loop->GetIndVarValue(), in ForWithStatus()
53 loop->GetIndVarValue(), start))); in ForWithStatus()
Dloop_emitter.cc108 array_multi_index[dimension] = loop->GetIndVarValue(); in EmitStaticIndex()
127 array_multi_index[dimension] = loop->GetIndVarValue(); in EmitDynamicIndex()
Dllvm_loop.h133 llvm::Value* GetIndVarValue() const { return indvar_; } in GetIndVarValue() function
Dllvm_loop.cc252 multi_index[dimension] = loop->GetIndVarValue(); in AddLoopsForShapeOnDimensions()
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc2216 inner_loop->GetIndVarValue()); in EmitElementalDot()
2229 inner_loop->GetIndVarValue()); in EmitElementalDot()
2859 ->GetIndVarValue(); in EmitConvolution()
2865 ->GetIndVarValue(); in EmitConvolution()