| /external/tensorflow/tensorflow/tools/benchmark/ |
| benchmark_model.cc | 53 std::unique_ptr<GraphDef>* graph_def) {
64 graph_def->reset(new GraphDef());
66 Status s = ReadBinaryProto(Env::Default(), graph, graph_def->get());
72 s = (*session)->Create(*(graph_def->get()));
460 std::unique_ptr<GraphDef> graph_def; local
464 InitializeSession(num_threads, graph, &session, &graph_def);
576 Status flop_status = CalculateFlops(*graph_def, inputs, session.get(),
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| quantize_nodes_test.cc | 834 GraphDef graph_def; local
859 GraphDef graph_def; local
[all...] |
| transform_utils_test.cc | 52 GraphDef graph_def; local
53 TF_ASSERT_OK(root.ToGraphDef(&graph_def));
55 MapNamesToNodes(graph_def, &node_map);
85 GraphDef graph_def; local
86 TF_ASSERT_OK(root.ToGraphDef(&graph_def));
89 MapNodesToOutputs(graph_def, &outputs_map);
241 GraphDef graph_def; local
242 TF_ASSERT_OK(root.ToGraphDef(&graph_def));
246 graph_def,
266 GraphDef graph_def; local
299 GraphDef graph_def; local
392 GraphDef graph_def; local
429 GraphDef graph_def; local
538 GraphDef graph_def; local
576 GraphDef graph_def; local
614 GraphDef graph_def; local
649 GraphDef graph_def; local
687 GraphDef graph_def; local
703 GraphDef graph_def; local
813 GraphDef graph_def; local
[all...] |
| /external/tensorflow/tensorflow/c/ |
| c_api_test.cc | 520 GraphDef graph_def; local
521 ASSERT_TRUE(GetGraphDef(graph, &graph_def));
527 for (const auto& n : graph_def.node()) {
561 NodeDef* added_node = graph_def.add_node();
563 EXPECT_EQ(ProtoDebugString(graph_def), ProtoDebugString(graph_def2));
651 TF_Buffer* graph_def = TF_NewBuffer(); local
652 TF_GraphToGraphDef(graph, graph_def, s);
660 TF_GraphImportGraphDef(graph, graph_def, opts, s);
707 TF_GraphImportGraphDefWithResults(graph, graph_def, opts, s);
749 TF_GraphImportGraphDef(graph, graph_def, opts, s)
828 TF_Buffer* graph_def = TF_NewBuffer(); local
879 TF_Buffer* graph_def = TF_NewBuffer(); local
[all...] |
| c_api.cc | 2538 GraphDef graph_def; local
[all...] |
