Searched refs:oned_shape (Results 1 – 1 of 1) sorted by relevance
938 const Shape oned_shape = ShapeUtil::MakeShape(element_type, {3}); in TEST_F() local942 ASSERT_TRUE(ShapeUtil::Equal(oned_shape, status.ValueOrDie())); in TEST_F()945 auto status = ShapeInference::InferBroadcastShape(oned_shape, {}); in TEST_F()947 ASSERT_TRUE(ShapeUtil::Equal(oned_shape, status.ValueOrDie())); in TEST_F()956 auto status = ShapeInference::InferBroadcastShape(oned_shape, {2}); in TEST_F()