| 30 | class QueryInferredShapeTest : public ClientLibraryTestBase {}; |
| 31 | |
| 32 | TEST_F(QueryInferredShapeTest, OnePlusOneShape) { |
| 33 | XlaBuilder builder("one_plus_one"); |
| 34 | auto one = ConstantR0<float>(&builder, 1.0); |
| 35 | auto result = Add(one, one); |
| 36 | StatusOr<Shape> shape_status = builder.GetShape(result); |
| 37 | ASSERT_IS_OK(shape_status.status()); |
| 38 | auto shape = shape_status.ConsumeValueOrDie(); |
| 39 | ASSERT_TRUE(ShapeUtil::Equal(shape, ShapeUtil::MakeShape(F32, {}))); |
| 40 | } |
| 41 | |
| 42 | } // namespace |
| 43 | } // namespace xla |