Searched refs:MakeDescendingLayout (Results 1 – 7 of 7) sorted by relevance
297 EXPECT_TRUE(LayoutUtil::Equal(LayoutUtil::MakeDescendingLayout(5), in TEST_F()299 EXPECT_TRUE(LayoutUtil::Equal(LayoutUtil::MakeDescendingLayout(1), in TEST_F()301 EXPECT_TRUE(LayoutUtil::Equal(LayoutUtil::MakeDescendingLayout(0), in TEST_F()
50 static Layout MakeDescendingLayout(int64_t rank);
84 /* static */ Layout LayoutUtil::MakeDescendingLayout(int64_t rank) { in MakeDescendingLayout() function in xla::LayoutUtil
354 LayoutUtil::MakeDescendingLayout( in AssertCorrectLayoutForDotOutputFusion()359 LayoutUtil::MakeDescendingLayout( in AssertCorrectLayoutForDotOutputFusion()
293 xla::LayoutUtil::MakeDescendingLayout(shape.dimensions().size())) { in CreateOpInFusion()518 xla::LayoutUtil::MakeDescendingLayout(shape.dimensions().size())) { in RewriteFusionOperand()
1564 LayoutUtil::MakeDescendingLayout( in ImportInstructionWithLayout()
267 return xla::LayoutUtil::MakeDescendingLayout(rank); in ExtractLayout()