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.cc345 if (!SameDefinedShape(node_context, node_context->input(dst_input), in UpdateNode()
352 if (!SameDefinedShape(node_context, node_context->input(dst_input), in UpdateNode()
724 bool ShapeRefiner::SameDefinedShape(InferenceContext* c, ShapeHandle s0, in SameDefinedShape() function in tensorflow::ShapeRefiner
755 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()
1266 TEST_F(ShapeRefinerTest, SameDefinedShape) { in TEST_F() argument
1284 EXPECT_TRUE(SameDefinedShape(ctx, unknown, unknown)); in TEST_F()
1285 EXPECT_FALSE(SameDefinedShape(ctx, unknown, unknown_b)); in TEST_F()
1286 EXPECT_FALSE(SameDefinedShape(ctx, unknown, s_1_2)); in TEST_F()
1287 EXPECT_TRUE(SameDefinedShape(ctx, s_1_2, s_1_2_b)); in TEST_F()
1288 EXPECT_FALSE(SameDefinedShape(ctx, s_1_2, s_2_2)); in TEST_F()
1289 EXPECT_TRUE(SameDefinedShape(ctx, s_unknown_2, s_unknown_2)); in TEST_F()
1290 EXPECT_FALSE(SameDefinedShape(ctx, s_unknown_2, s_unknown_2_b)); in TEST_F()
Dshape_refiner.h191 static bool SameDefinedShape(shape_inference::InferenceContext* c,