| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | loop_emitter.cc | 102 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()
|
| D | loop_emitter.h | 103 IrArray::Index EmitStaticIndex(ForLoopNest* loop_nest,
105 IrArray::Index EmitDynamicIndex(ForLoopNest* loop_nest,
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | parallel_loop_emitter.cc | 46 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()
|
| D | dot_op_emitter.cc | 638 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()
|
| D | ir_emitter.cc | 1804 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/spirv-tools/src/test/opt/loop_optimizations/ |
| D | dependence_analysis.cpp | 3062 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/swiftshader/third_party/SPIRV-Tools/test/opt/loop_optimizations/ |
| D | dependence_analysis.cpp | 3062 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/ |
| D | dependence_analysis.cpp | 3062 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/pytorch/torch/_inductor/codegen/ |
| D | cpp.py | 435 left_loop_nest = cpp_kernel_proxy_list[idx].loop_nest
436 right_loop_nest = cpp_kernel_proxy_list[idx + 1].loop_nest
439 len(loop_nest.root) != 1
440 for loop_nest in [left_loop_nest, right_loop_nest]
453 kernel.loop_nest for kernel in cpp_kernel_proxy_list
1978 def codegen_loops_impl(self, loop_nest, code, worksharing): argument
1981 kernels = loop_nest.get_kernels()
1989 loop_nest.max_parallel_depth(), threads
1993 loop_nest.max_parallel_depth(), threads
1998 if loop_nest.is_reduction_only():
[all …]
|
| /external/pytorch/test/cpp/tensorexpr/ |
| D | test_llvm.cpp | 1527 LoopNest loop_nest({f}); in testSimpleParallel() local
1528 auto const& loops = loop_nest.getLoopStmtsFor(f); in testSimpleParallel()
1537 loop_nest.prepareForCodegen(); in testSimpleParallel()
1538 StmtPtr stmt = loop_nest.root_stmt(); in testSimpleParallel()
1585 LoopNest loop_nest({t4}, {t1, t2, t3, t4}); in TEST() local
1588 auto const& loops = loop_nest.getLoopStmtsFor(t1); in TEST()
1592 auto const& loops = loop_nest.getLoopStmtsFor(t2); in TEST()
1596 auto const& loops = loop_nest.getLoopStmtsFor(t3); in TEST()
1601 auto const& loops = loop_nest.getLoopStmtsFor(t4); in TEST()
1611 loop_nest.prepareForCodegen(); in TEST()
[all …]
|