Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dparallel_loop_emitter.cc57 array_multi_index[dimension] = loop->GetIndVarValue(); in EmitIndexAndSetExitBasicBlock()
64 array_multi_index[dimension] = loop->GetIndVarValue(); in EmitIndexAndSetExitBasicBlock()
Ddot_op_emitter.cc457 lhs_multi_index[lhs_reduction_dimension] = reduction_loop->GetIndVarValue(); in EmitNaiveLlvmIrGemm()
460 rhs_multi_index[rhs_reduction_dimension] = reduction_loop->GetIndVarValue(); in EmitNaiveLlvmIrGemm()
Dir_emitter.cc988 ->GetIndVarValue(); in EmitElementalConvolution()
994 ->GetIndVarValue(); in EmitElementalConvolution()
1732 array_multi_index[dimension] = loop->GetIndVarValue(); in EmitVectorizedReduce()
1753 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 EmitIndexAndSetExitBasicBlock()
Dllvm_loop.h133 llvm::Value* GetIndVarValue() const { return indvar_; } in GetIndVarValue() function
Dllvm_loop.cc251 multi_index[dimension] = loop->GetIndVarValue(); in AddLoopsForShapeOnDimensions()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc553 lhs_multi_index[lhs_reduction_dimension] = reduction_loop->GetIndVarValue(); in HandleDot()
554 rhs_multi_index[rhs_reduction_dimension] = reduction_loop->GetIndVarValue(); in HandleDot()
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc2157 inner_loop->GetIndVarValue()); in EmitElementalDot()
2170 inner_loop->GetIndVarValue()); in EmitElementalDot()