Home
last modified time | relevance | path

Searched refs:merged_shapes (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dshape_inference_test.cc672 auto merged_shapes = c.MergedShapes(); in TEST_F() local
673 ASSERT_EQ(3, merged_shapes.size()); in TEST_F()
674 EXPECT_TRUE(merged_shapes[0].first.SameHandle(s_unknown)); in TEST_F()
675 EXPECT_TRUE(merged_shapes[0].second.SameHandle(s_1_2)); in TEST_F()
676 EXPECT_TRUE(merged_shapes[1].first.SameHandle(s_u_2)); in TEST_F()
677 EXPECT_TRUE(merged_shapes[1].second.SameHandle(s_unknown)); in TEST_F()
678 EXPECT_TRUE(merged_shapes[2].first.SameHandle(s_unknown)); in TEST_F()
679 EXPECT_TRUE(merged_shapes[2].second.SameHandle(s_unknown_b)); in TEST_F()
685 merged_shapes = c.MergedShapes(); in TEST_F()
686 EXPECT_EQ(3, merged_shapes.size()); in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc2154 for (const auto& merged_shapes : node_ctx->MergedShapes()) { in InferStatically() local
2155 if (!shape_manager->Merge(merged_shapes.first, merged_shapes.second) in InferStatically()