Searched refs:fused_loop (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/test/cpp/tensorexpr/ |
| D | test_loopnest.cpp | 5353 ForPtr fused_loop; in TEST() local
5354 ASSERT_TRUE(LoopNest::fuseLoops({forJ, forK}, &fused_loop)); in TEST()
5368 ASSERT_EQ(fused_loop, forJ); in TEST()
5393 ForPtr fused_loop; in TEST() local
5394 ASSERT_TRUE(LoopNest::fuseLoops({forI, forJ, forK}, &fused_loop)); in TEST()
5409 ASSERT_EQ(fused_loop, forI); in TEST()
5450 ForPtr fused_loop; in TEST() local
5451 ASSERT_TRUE(LoopNest::fuseLoops({forM, forN}, &fused_loop)); in TEST()
5469 ASSERT_EQ(fused_loop, forM); in TEST()
5510 ForPtr fused_loop; in TEST() local
[all …]
|
| /external/pytorch/torch/csrc/jit/tensorexpr/ |
| D | tensorexpr_init.cpp | 598 ForPtr fused_loop = nullptr; in initTensorExprBindings() local
599 LoopNest::unsafeFuseLoops(loops, &fused_loop); in initTensorExprBindings()
600 return fused_loop; in initTensorExprBindings()
606 ForPtr fused_loop = nullptr; in initTensorExprBindings() local
607 LoopNest::fuseLoops(loops, &fused_loop); in initTensorExprBindings()
608 return fused_loop; in initTensorExprBindings()
|
| D | loopnest_randomization.cpp | 382 ForPtr fused_loop; in loopnestRandomization() local
383 l.fuseLoops(loops_to_fuse, &fused_loop); in loopnestRandomization()
|