Searched refs:areLoopsPerfectlyNested (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/torch/csrc/jit/tensorexpr/ |
| D | loopnest.h | 418 static bool areLoopsPerfectlyNested(const std::vector<ForPtr>& loops);
|
| D | loopnest.cpp | 2181 if (!areLoopsPerfectlyNested(loops)) { in reorder()
2254 if (!areLoopsPerfectlyNested({x, y})) { in tile()
2285 bool LoopNest::areLoopsPerfectlyNested(const std::vector<ForPtr>& loops) { in areLoopsPerfectlyNested() function in torch::jit::tensorexpr::LoopNest
|
| /external/pytorch/test/cpp/tensorexpr/ |
| D | test_loopnest.cpp | 6269 TEST(LoopNest, areLoopsPerfectlyNested) { in TEST() argument
6288 ASSERT_TRUE(LoopNest::areLoopsPerfectlyNested({forI, forJ, forK})); in TEST()
6291 ASSERT_FALSE(LoopNest::areLoopsPerfectlyNested({forJ, forI, forK})); in TEST()
6292 ASSERT_FALSE(LoopNest::areLoopsPerfectlyNested({forI, forK, forJ})); in TEST()
6293 ASSERT_FALSE(LoopNest::areLoopsPerfectlyNested({forK, forJ, forI})); in TEST()
6298 ASSERT_TRUE(LoopNest::areLoopsPerfectlyNested({forI, forJ, forK})); in TEST()
6303 ASSERT_FALSE(LoopNest::areLoopsPerfectlyNested({forI, forJ, forK})); in TEST()
6308 ASSERT_FALSE(LoopNest::areLoopsPerfectlyNested({forI, forJ, forK})); in TEST()
6330 ASSERT_TRUE(LoopNest::areLoopsPerfectlyNested({forJ, forI})); in TEST()
6359 ASSERT_TRUE(LoopNest::areLoopsPerfectlyNested({forK, forI, forJ})); in TEST()
[all …]
|