Home
last modified time | relevance | path

Searched refs:SameDefinedShape (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc323 if (!SameDefinedShape(node_context, node_context->input(dst_input), in UpdateNode()
330 if (!SameDefinedShape(node_context, node_context->input(dst_input), in UpdateNode()
767 bool ShapeRefiner::SameDefinedShape(InferenceContext* c, ShapeHandle s0, in SameDefinedShape() function in tensorflow::ShapeRefiner
798 if (!SameDefinedShape(c, existing[i].shape, updated[i].shape) || in IsUpdatedShapesOrTypes()
Dshape_refiner_test.cc49 bool SameDefinedShape(shape_inference::InferenceContext* c, in SameDefinedShape() function in tensorflow::ShapeRefinerTest
52 return ShapeRefiner::SameDefinedShape(c, s0, s1); in SameDefinedShape()
1251 TEST_F(ShapeRefinerTest, SameDefinedShape) { in TEST_F() argument
1269 EXPECT_TRUE(SameDefinedShape(ctx, unknown, unknown)); in TEST_F()
1270 EXPECT_FALSE(SameDefinedShape(ctx, unknown, unknown_b)); in TEST_F()
1271 EXPECT_FALSE(SameDefinedShape(ctx, unknown, s_1_2)); in TEST_F()
1272 EXPECT_TRUE(SameDefinedShape(ctx, s_1_2, s_1_2_b)); in TEST_F()
1273 EXPECT_FALSE(SameDefinedShape(ctx, s_1_2, s_2_2)); in TEST_F()
1274 EXPECT_TRUE(SameDefinedShape(ctx, s_unknown_2, s_unknown_2)); in TEST_F()
1275 EXPECT_FALSE(SameDefinedShape(ctx, s_unknown_2, s_unknown_2_b)); in TEST_F()
Dshape_refiner.h162 static bool SameDefinedShape(shape_inference::InferenceContext* c,