Searched refs:merged_shapes (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference_test.cc | 687 auto merged_shapes = c.MergedShapes(); in TEST_F() local
688 ASSERT_EQ(3, merged_shapes.size()); in TEST_F()
689 EXPECT_TRUE(merged_shapes[0].first.SameHandle(s_unknown)); in TEST_F()
690 EXPECT_TRUE(merged_shapes[0].second.SameHandle(s_1_2)); in TEST_F()
691 EXPECT_TRUE(merged_shapes[1].first.SameHandle(s_u_2)); in TEST_F()
692 EXPECT_TRUE(merged_shapes[1].second.SameHandle(s_unknown)); in TEST_F()
693 EXPECT_TRUE(merged_shapes[2].first.SameHandle(s_unknown)); in TEST_F()
694 EXPECT_TRUE(merged_shapes[2].second.SameHandle(s_unknown_b)); in TEST_F()
700 merged_shapes = c.MergedShapes(); in TEST_F()
701 EXPECT_EQ(3, merged_shapes.size()); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 2170 std::string merged_shapes = "Merged shapes from SymbolicShapManager:\n"; in VerboseShapeInferenceLogging() local
2173 &merged_shapes, " input[", i, "] -- ", in VerboseShapeInferenceLogging()
2179 &merged_shapes, " output[", i, "] -- ", in VerboseShapeInferenceLogging()
2183 VLOG(3) << merged_shapes; in VerboseShapeInferenceLogging()
2591 for (const auto& merged_shapes : node_ctx->MergedShapes()) { in InferStatically() local
2592 if (!shape_manager->Merge(merged_shapes.first, merged_shapes.second) in InferStatically()
|