Home
last modified time | relevance | path

Searched refs:GetGraphDef (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/
Dtrt_convert_api_test.cc50 GraphDef GetGraphDef(PartialTensorShape input_shape) { in GetGraphDef() function in tensorflow::tensorrt::TrtConverterTest
161 *(out.mutable_graph_def()) = GetGraphDef(shape); in GetModel()
/external/tensorflow/tensorflow/c/
Dc_test_util.h116 bool GetGraphDef(TF_Graph* graph, tensorflow::GraphDef* graph_def);
Dc_api_function_test.cc1331 GetGraphDef(host_graph_, &gdef); in TEST_F()
1353 GetGraphDef(host_graph_, &gdef2); in TEST_F()
1379 GetGraphDef(host_graph_, &gdef); in TEST_F()
1407 GetGraphDef(host_graph_, &gdef); in TEST_F()
1420 GetGraphDef(host_graph_, &gdef); in TEST_F()
Dc_test_util.cc388 bool GetGraphDef(TF_Graph* graph, tensorflow::GraphDef* graph_def) { in GetGraphDef() function
Dc_api_test.cc548 ASSERT_TRUE(GetGraphDef(graph, &graph_def)); in TEST()
585 ASSERT_TRUE(GetGraphDef(graph, &graph_def2)); in TEST()
1616 EXPECT_TRUE(GetGraphDef(expected_graph_, &expected_gdef)); in TestGradientsSuccess()
1617 EXPECT_TRUE(GetGraphDef(graph_, &gdef)); in TestGradientsSuccess()