Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dparallel_loop_emitter.cc40 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()
Ddot_op_emitter.cc431 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()
Dir_emitter.cc1722 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/
Dloop_emitter.cc100 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/
Ddependence_analysis.cpp3068 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/
Ddependence_analysis.cpp3068 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/
Dir_emitter.cc530 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()