Searched defs:loop_nest (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | loop_emitter.cc | 102 IrArray::Index LoopEmitter::EmitStaticIndex(ForLoopNest* loop_nest, in EmitStaticIndex()
120 IrArray::Index LoopEmitter::EmitDynamicIndex(ForLoopNest* loop_nest, in EmitDynamicIndex()
154 ForLoopNest loop_nest(loop_name, b_); in EmitIndexAndSetExitBasicBlock() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | parallel_loop_emitter.cc | 46 llvm_ir::ForLoopNest loop_nest(loop_name, b_); in EmitIndexAndSetExitBasicBlock() local
|
| D | dot_op_emitter.cc | 638 llvm_ir::ForLoopNest loop_nest(llvm_ir::IrName(dot_hlo_name_), b_); in EmitNaiveLlvmIrGemm() local
|
| D | ir_emitter.cc | 1804 llvm_ir::ForLoopNest loop_nest(IrName(reduce), &b_); in EmitVectorizedReduce() local
|
| /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
3470 std::vector<const Loop*> loop_nest{&ld.GetLoopByIndex(0), in TEST() local
3511 std::vector<const Loop*> loop_nest{&ld.GetLoopByIndex(0)}; in TEST() local
3543 std::vector<const Loop*> loop_nest{&ld.GetLoopByIndex(0)}; in TEST() local
3578 std::vector<const Loop*> loop_nest{ in TEST() local
3620 std::vector<const Loop*> loop_nest{&ld.GetLoopByIndex(0), in TEST() local
|
| /external/swiftshader/third_party/SPIRV-Tools/test/opt/loop_optimizations/ |
| D | dependence_analysis.cpp | 3062 std::vector<const Loop*> loop_nest{ in TEST() local
3470 std::vector<const Loop*> loop_nest{&ld.GetLoopByIndex(0), in TEST() local
3511 std::vector<const Loop*> loop_nest{&ld.GetLoopByIndex(0)}; in TEST() local
3543 std::vector<const Loop*> loop_nest{&ld.GetLoopByIndex(0)}; in TEST() local
3578 std::vector<const Loop*> loop_nest{ in TEST() local
3620 std::vector<const Loop*> loop_nest{&ld.GetLoopByIndex(0), in TEST() local
|
| /external/deqp-deps/SPIRV-Tools/test/opt/loop_optimizations/ |
| D | dependence_analysis.cpp | 3062 std::vector<const Loop*> loop_nest{ in TEST() local
3470 std::vector<const Loop*> loop_nest{&ld.GetLoopByIndex(0), in TEST() local
3511 std::vector<const Loop*> loop_nest{&ld.GetLoopByIndex(0)}; in TEST() local
3543 std::vector<const Loop*> loop_nest{&ld.GetLoopByIndex(0)}; in TEST() local
3578 std::vector<const Loop*> loop_nest{ in TEST() local
3620 std::vector<const Loop*> loop_nest{&ld.GetLoopByIndex(0), in TEST() local
|
| /external/pytorch/test/cpp/tensorexpr/ |
| D | test_llvm.cpp | 1527 LoopNest loop_nest({f}); in testSimpleParallel() local
1585 LoopNest loop_nest({t4}, {t1, t2, t3, t4}); in TEST() local
|
| /external/pytorch/torch/_inductor/codegen/ |
| D | cpp.py | 1978 def codegen_loops_impl(self, loop_nest, code, worksharing): argument
|