Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dloop_emitter.cc102 IrArray::Index LoopEmitter::EmitStaticIndex(ForLoopNest* loop_nest, in EmitStaticIndex() argument
111 std::unique_ptr<ForLoop> loop = loop_nest->AddLoop( in EmitStaticIndex()
120 IrArray::Index LoopEmitter::EmitDynamicIndex(ForLoopNest* loop_nest, in EmitDynamicIndex() argument
130 std::unique_ptr<ForLoop> loop = loop_nest->AddLoop( in EmitDynamicIndex()
154 ForLoopNest loop_nest(loop_name, b_); in EmitIndexAndSetExitBasicBlock() local
157 ? EmitStaticIndex(&loop_nest, index_type) in EmitIndexAndSetExitBasicBlock()
158 : EmitDynamicIndex(&loop_nest, index_type); in EmitIndexAndSetExitBasicBlock()
162 llvm::BasicBlock* innermost_body_bb = loop_nest.GetInnerLoopBodyBasicBlock(); in EmitIndexAndSetExitBasicBlock()
167 exit_bb_ = loop_nest.GetOuterLoopExitBasicBlock(); in EmitIndexAndSetExitBasicBlock()
Dloop_emitter.h103 IrArray::Index EmitStaticIndex(ForLoopNest* loop_nest,
105 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.cc638 llvm_ir::ForLoopNest loop_nest(llvm_ir::IrName(dot_hlo_name_), b_); in EmitNaiveLlvmIrGemm() local
640 loop_nest.EmitOperandArrayLoopNest( in EmitNaiveLlvmIrGemm()
643 loop_nest.EmitOperandArrayLoopNest( in EmitNaiveLlvmIrGemm()
654 std::unique_ptr<llvm_ir::ForLoop> reduction_loop = loop_nest.AddLoop( in EmitNaiveLlvmIrGemm()
756 b_->SetInsertPoint(loop_nest.GetOuterLoopExitBasicBlock()); in EmitNaiveLlvmIrGemm()
Dir_emitter.cc1804 llvm_ir::ForLoopNest loop_nest(IrName(reduce), &b_); in EmitVectorizedReduce() local
1812 std::unique_ptr<llvm_ir::ForLoop> loop = loop_nest.AddLoop( in EmitVectorizedReduce()
1822 loop_nest.GetInnerLoopBodyBasicBlock()) { in EmitVectorizedReduce()
1826 auto outermost_loop_exit_block = loop_nest.GetOuterLoopExitBasicBlock(); in EmitVectorizedReduce()
1833 loop_nest.AddLoop(start_index, end_index, vectorization_factor, in EmitVectorizedReduce()
/external/angle/third_party/vulkan-deps/spirv-tools/src/test/opt/loop_optimizations/
Ddependence_analysis.cpp3062 std::vector<const Loop*> loop_nest{ in TEST() local
3065 LoopDependenceAnalysis analysis{context.get(), loop_nest}; in TEST()
3470 std::vector<const Loop*> loop_nest{&ld.GetLoopByIndex(0), in TEST() local
3472 LoopDependenceAnalysis analysis{context.get(), loop_nest}; in TEST()
3474 DistanceVector dv_entry(loop_nest.size()); in TEST()
3511 std::vector<const Loop*> loop_nest{&ld.GetLoopByIndex(0)}; in TEST() local
3512 LoopDependenceAnalysis analysis{context.get(), loop_nest}; in TEST()
3514 DistanceVector dv_entry(loop_nest.size()); in TEST()
3543 std::vector<const Loop*> loop_nest{&ld.GetLoopByIndex(0)}; in TEST() local
3544 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 …]
/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 …]