Searched refs:ShapeWithLayout (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_matchers_test.cc | 115 ::testing::Not(op::ShapeWithLayout(ShapeUtil::MakeShape(F32, {5, 7})))); in TEST()
116 EXPECT_THAT(p0.get(), ::testing::Not(op::ShapeWithLayout("f32[5,7]"))); in TEST()
122 ::testing::Not(op::ShapeWithLayout(ShapeUtil::MakeShape(F32, {7, 5})))); in TEST()
123 EXPECT_THAT(p0.get(), ::testing::Not(op::ShapeWithLayout("f32[7,5]"))); in TEST()
127 EXPECT_THAT(p0.get(), op::ShapeWithLayout(ShapeUtil::MakeShapeWithLayout( in TEST()
129 EXPECT_THAT(p0.get(), op::ShapeWithLayout("f32[5,7]{0,1}")); in TEST()
131 ::testing::Not(op::ShapeWithLayout( in TEST()
133 EXPECT_THAT(p0.get(), ::testing::Not(op::ShapeWithLayout("f32[5,7]{1,0}"))); in TEST()
139 Explain(p0.get(), op::ShapeWithLayout(ShapeUtil::MakeShapeWithLayout( in TEST()
|
| D | hlo_matchers.h | 360 inline ::testing::Matcher<const ::xla::HloInstruction*> ShapeWithLayout( in ShapeWithLayout() function
365 inline ::testing::Matcher<const ::xla::HloInstruction*> ShapeWithLayout( in ShapeWithLayout() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment_test.cc | 364 op::Dot(op::ShapeWithLayout(expected_shape), in TEST_F()
365 op::ShapeWithLayout(expected_shape))); in TEST_F()
401 op::Sort(op::ShapeWithLayout(expected_shape), in TEST_F()
402 op::ShapeWithLayout(expected_shape))); in TEST_F()
|