Searched refs:loop_nest (Results 1 – 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | parallel_loop_emitter.cc | 40 llvm_ir::ForLoopNest loop_nest(loop_name, b_); in EmitIndexAndSetExitBasicBlock() local 54 std::unique_ptr<llvm_ir::ForLoop> loop = loop_nest.AddLoop( in EmitIndexAndSetExitBasicBlock() 60 std::unique_ptr<llvm_ir::ForLoop> loop = loop_nest.AddLoop( in EmitIndexAndSetExitBasicBlock() 68 llvm_ir::SetToFirstInsertPoint(loop_nest.GetInnerLoopBodyBasicBlock(), b_); in EmitIndexAndSetExitBasicBlock() 71 exit_bb_ = loop_nest.GetOuterLoopExitBasicBlock(); in EmitIndexAndSetExitBasicBlock()
|
D | dot_op_emitter.cc | 431 llvm_ir::ForLoopNest loop_nest(llvm_ir::IrName(dot_hlo_name_), b_); in EmitNaiveLlvmIrGemm() local 433 loop_nest.EmitOperandArrayLoopNest( in EmitNaiveLlvmIrGemm() 436 loop_nest.EmitOperandArrayLoopNest( in EmitNaiveLlvmIrGemm() 447 std::unique_ptr<llvm_ir::ForLoop> reduction_loop = loop_nest.AddLoop( in EmitNaiveLlvmIrGemm() 543 b_->SetInsertPoint(loop_nest.GetOuterLoopExitBasicBlock()); in EmitNaiveLlvmIrGemm()
|
D | ir_emitter.cc | 1722 llvm_ir::ForLoopNest loop_nest(IrName(reduce), &b_); in EmitVectorizedReduce() local 1730 std::unique_ptr<llvm_ir::ForLoop> loop = loop_nest.AddLoop( in EmitVectorizedReduce() 1740 loop_nest.GetInnerLoopBodyBasicBlock()) { in EmitVectorizedReduce() 1744 auto outermost_loop_exit_block = loop_nest.GetOuterLoopExitBasicBlock(); in EmitVectorizedReduce() 1751 loop_nest.AddLoop(start_index, end_index, vectorization_factor, in EmitVectorizedReduce()
|
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
D | loop_emitter.cc | 100 ForLoopNest loop_nest(loop_name, b_); in EmitIndexAndSetExitBasicBlock() local 104 std::unique_ptr<ForLoop> loop = loop_nest.AddLoop( in EmitIndexAndSetExitBasicBlock() 114 llvm::BasicBlock* innermost_body_bb = loop_nest.GetInnerLoopBodyBasicBlock(); in EmitIndexAndSetExitBasicBlock() 119 exit_bb_ = loop_nest.GetOuterLoopExitBasicBlock(); in EmitIndexAndSetExitBasicBlock()
|
/external/deqp-deps/SPIRV-Tools/test/opt/loop_optimizations/ |
D | dependence_analysis.cpp | 3068 std::vector<const Loop*> loop_nest{ in TEST() local 3071 LoopDependenceAnalysis analysis{context.get(), loop_nest}; in TEST() 3476 std::vector<const Loop*> loop_nest{&ld.GetLoopByIndex(0), in TEST() local 3478 LoopDependenceAnalysis analysis{context.get(), loop_nest}; in TEST() 3480 DistanceVector dv_entry(loop_nest.size()); in TEST() 3517 std::vector<const Loop*> loop_nest{&ld.GetLoopByIndex(0)}; in TEST() local 3518 LoopDependenceAnalysis analysis{context.get(), loop_nest}; in TEST() 3520 DistanceVector dv_entry(loop_nest.size()); in TEST() 3549 std::vector<const Loop*> loop_nest{&ld.GetLoopByIndex(0)}; in TEST() local 3550 LoopDependenceAnalysis analysis{context.get(), loop_nest}; in TEST() [all …]
|
/external/swiftshader/third_party/SPIRV-Tools/test/opt/loop_optimizations/ |
D | dependence_analysis.cpp | 3068 std::vector<const Loop*> loop_nest{ in TEST() local 3071 LoopDependenceAnalysis analysis{context.get(), loop_nest}; in TEST() 3476 std::vector<const Loop*> loop_nest{&ld.GetLoopByIndex(0), in TEST() local 3478 LoopDependenceAnalysis analysis{context.get(), loop_nest}; in TEST() 3480 DistanceVector dv_entry(loop_nest.size()); in TEST() 3517 std::vector<const Loop*> loop_nest{&ld.GetLoopByIndex(0)}; in TEST() local 3518 LoopDependenceAnalysis analysis{context.get(), loop_nest}; in TEST() 3520 DistanceVector dv_entry(loop_nest.size()); in TEST() 3549 std::vector<const Loop*> loop_nest{&ld.GetLoopByIndex(0)}; in TEST() local 3550 LoopDependenceAnalysis analysis{context.get(), loop_nest}; in TEST() [all …]
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | ir_emitter.cc | 530 llvm_ir::ForLoopNest loop_nest(IrName(dot), &b_); in HandleDot() local 532 loop_nest.EmitOperandArrayLoopNest( in HandleDot() 535 loop_nest.EmitOperandArrayLoopNest( in HandleDot() 546 std::unique_ptr<llvm_ir::ForLoop> reduction_loop = loop_nest.AddLoop( in HandleDot() 632 b_.SetInsertPoint(loop_nest.GetOuterLoopExitBasicBlock()); in HandleDot()
|