Searched refs:static_graph_properties (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| D | segment_test.cc | 391 grappler::GraphProperties static_graph_properties(item); in TEST_F() local
392 TF_EXPECT_OK(static_graph_properties.InferStatically(true)); in TEST_F()
400 RunTest(&g, &static_graph_properties, all_nodes, all_nodes, all_nodes, in TEST_F()
423 grappler::GraphProperties static_graph_properties(item); in TEST_F() local
424 TF_EXPECT_OK(static_graph_properties.InferStatically(true)); in TEST_F()
432 RunTest(&g, &static_graph_properties, all_nodes, all_nodes, all_nodes, {}); in TEST_F()
452 grappler::GraphProperties static_graph_properties(item); in TEST_F() local
453 TF_EXPECT_OK(static_graph_properties.InferStatically(true)); in TEST_F()
461 RunTest(&g, &static_graph_properties, all_nodes, all_nodes, all_nodes, {}); in TEST_F()
481 grappler::GraphProperties static_graph_properties(item); in TEST_F() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 719 grappler::GraphProperties static_graph_properties(grappler_item); in ConvertAfterShapes() local
720 TF_RETURN_IF_ERROR(static_graph_properties.InferStatically(true)); in ConvertAfterShapes()
742 TrtNodeValidator validator(static_graph_properties, params.precision_mode, in ConvertAfterShapes()
745 &graph, &static_graph_properties, in ConvertAfterShapes()
770 Status status = GetEngineInfo(&graph, static_graph_properties, curr_segment, in ConvertAfterShapes()
|