Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dloop_emitter.cc95 IrArray::Index LoopEmitter::EmitStaticIndex(ForLoopNest* loop_nest, in EmitStaticIndex() argument
104 std::unique_ptr<ForLoop> loop = loop_nest->AddLoop( in EmitStaticIndex()
113 IrArray::Index LoopEmitter::EmitDynamicIndex(ForLoopNest* loop_nest, in EmitDynamicIndex() argument
123 std::unique_ptr<ForLoop> loop = loop_nest->AddLoop( in EmitDynamicIndex()
147 ForLoopNest loop_nest(loop_name, b_); in EmitIndexAndSetExitBasicBlock() local
150 ? EmitStaticIndex(&loop_nest, index_type) in EmitIndexAndSetExitBasicBlock()
151 : EmitDynamicIndex(&loop_nest, index_type); in EmitIndexAndSetExitBasicBlock()
155 llvm::BasicBlock* innermost_body_bb = loop_nest.GetInnerLoopBodyBasicBlock(); in EmitIndexAndSetExitBasicBlock()
160 exit_bb_ = loop_nest.GetOuterLoopExitBasicBlock(); in EmitIndexAndSetExitBasicBlock()
Dloop_emitter.h104 IrArray::Index EmitStaticIndex(ForLoopNest* loop_nest,
106 IrArray::Index EmitDynamicIndex(ForLoopNest* loop_nest,
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dparallel_loop_emitter.cc46 llvm_ir::ForLoopNest loop_nest(loop_name, b_); in EmitIndexAndSetExitBasicBlock() local
60 std::unique_ptr<llvm_ir::ForLoop> loop = loop_nest.AddLoop( in EmitIndexAndSetExitBasicBlock()
66 std::unique_ptr<llvm_ir::ForLoop> loop = loop_nest.AddLoop( in EmitIndexAndSetExitBasicBlock()
74 llvm_ir::SetToFirstInsertPoint(loop_nest.GetInnerLoopBodyBasicBlock(), b_); in EmitIndexAndSetExitBasicBlock()
77 exit_bb_ = loop_nest.GetOuterLoopExitBasicBlock(); in EmitIndexAndSetExitBasicBlock()
Ddot_op_emitter.cc574 llvm_ir::ForLoopNest loop_nest(llvm_ir::IrName(dot_hlo_name_), b_); in EmitNaiveLlvmIrGemm() local
576 loop_nest.EmitOperandArrayLoopNest( in EmitNaiveLlvmIrGemm()
579 loop_nest.EmitOperandArrayLoopNest( in EmitNaiveLlvmIrGemm()
590 std::unique_ptr<llvm_ir::ForLoop> reduction_loop = loop_nest.AddLoop( in EmitNaiveLlvmIrGemm()
692 b_->SetInsertPoint(loop_nest.GetOuterLoopExitBasicBlock()); in EmitNaiveLlvmIrGemm()
Dir_emitter.cc1702 llvm_ir::ForLoopNest loop_nest(IrName(reduce), &b_); in EmitVectorizedReduce() local
1710 std::unique_ptr<llvm_ir::ForLoop> loop = loop_nest.AddLoop( in EmitVectorizedReduce()
1720 loop_nest.GetInnerLoopBodyBasicBlock()) { in EmitVectorizedReduce()
1724 auto outermost_loop_exit_block = loop_nest.GetOuterLoopExitBasicBlock(); in EmitVectorizedReduce()
1731 loop_nest.AddLoop(start_index, end_index, vectorization_factor, in EmitVectorizedReduce()
/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/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 …]