Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc654 std::unique_ptr<llvm_ir::ForLoop> reduction_loop = loop_nest.AddLoop( in EmitNaiveLlvmIrGemm() local
664 lhs_multi_index[lhs_reduction_dimension] = reduction_loop->GetIndVarValue(); in EmitNaiveLlvmIrGemm()
667 rhs_multi_index[rhs_reduction_dimension] = reduction_loop->GetIndVarValue(); in EmitNaiveLlvmIrGemm()
677 llvm::Function* func = reduction_loop->GetPreheaderBasicBlock()->getParent(); in EmitNaiveLlvmIrGemm()
685 llvm::BasicBlock* preheader_bb = reduction_loop->GetPreheaderBasicBlock(); in EmitNaiveLlvmIrGemm()
695 SetToFirstInsertPoint(reduction_loop->GetBodyBasicBlock(), b_); in EmitNaiveLlvmIrGemm()
730 SetToFirstInsertPoint(reduction_loop->GetExitBasicBlock(), b_); in EmitNaiveLlvmIrGemm()