Searched refs:LayoutOf (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment_test.cc | 62 std::vector<int64> LayoutOf(HloModule* m, absl::string_view name) { in LayoutOf() function in xla::__anona5bdbb650111::LayoutAssignmentTest
736 EXPECT_THAT(LayoutOf(m.get(), "gte0"), ElementsAre(0, 1, 2)); in TEST_F()
737 EXPECT_THAT(LayoutOf(m.get(), "gte1a"), ElementsAre(1, 2, 0)); in TEST_F()
738 EXPECT_THAT(LayoutOf(m.get(), "gte1b"), ElementsAre(2, 0, 1)); in TEST_F()
739 EXPECT_THAT(LayoutOf(m.get(), "fresult"), ElementsAre(2, 1, 0)); in TEST_F()
875 EXPECT_THAT(LayoutOf(m.get(), "gte"), ElementsAre(0, 1)); in TEST_F()
876 EXPECT_THAT(LayoutOf(m.get(), "root"), ElementsAre(1, 0)); in TEST_F()
919 EXPECT_THAT(LayoutOf(m.get(), "gte"), ElementsAre(0, 1)); in TEST_F()
920 EXPECT_THAT(LayoutOf(m.get(), "ar.0"), ElementsAre(0, 1)); in TEST_F()
921 EXPECT_THAT(LayoutOf(m.get(), "ar.1"), ElementsAre(0, 1)); in TEST_F()
[all …]
|