| /external/tensorflow/tensorflow/core/graph/ |
| graph_def_builder_test.cc | 43 GraphDef graph_def; local
44 TF_EXPECT_OK(builder.ToGraphDef(&graph_def));
45 ASSERT_EQ(graph_def.versions().producer(), TF_GRAPH_DEF_VERSION);
46 ASSERT_EQ(graph_def.versions().min_consumer(),
|
| graph_def_builder_util.cc | 22 GraphDef graph_def; local
23 TF_RETURN_IF_ERROR(builder.ToGraphDef(&graph_def));
25 return ConvertGraphDefToGraph(opts, graph_def, graph);
|
| validate_test.cc | 43 GraphDef graph_def; local
45 CHECK(parser.MergeFromString(graph_def_str, &graph_def)) << graph_def_str;
46 TF_ASSERT_OK(graph::ValidateGraphDef(graph_def, *OpRegistry::Global()));
58 GraphDef graph_def; local
60 CHECK(parser.MergeFromString(graph_def_str, &graph_def)) << graph_def_str;
61 Status s = graph::ValidateGraphDef(graph_def, *OpRegistry::Global());
66 TF_ASSERT_OK(AddDefaultAttrsToGraphDef(&graph_def, *OpRegistry::Global(), 0));
69 TF_ASSERT_OK(graph::ValidateGraphDef(graph_def, *OpRegistry::Global()));
81 GraphDef graph_def; local
83 CHECK(parser.MergeFromString(graph_def_str, &graph_def)) << graph_def_str
103 GraphDef graph_def; local
115 GraphDef graph_def; local
[all...] |
| /external/tensorflow/tensorflow/core/util/ |
| stat_summarizer_test.cc | 58 GraphDef graph_def; local
59 ASSERT_TRUE(protobuf::TextFormat::ParseFromString(graph_def_str, &graph_def));
63 TF_ASSERT_OK(session->Create(graph_def));
74 StatSummarizer stats(graph_def);
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| add_default_attributes_test.cc | 39 GraphDef graph_def; local
41 NodeDef* lrn_node1 = graph_def.add_node();
45 NodeDef* lrn_node2 = graph_def.add_node();
54 TF_ASSERT_OK(AddDefaultAttributes(graph_def, {}, &result));
|
| fake_quantize_training_test.cc | 51 GraphDef graph_def; local
52 TF_ASSERT_OK(root.ToGraphDef(&graph_def));
56 TF_ASSERT_OK(FakeQuantizeTraining(graph_def, context, &result));
59 EXPECT_GT(result.node_size(), graph_def.node_size());
|
| remove_device_test.cc | 39 GraphDef graph_def; local
41 NodeDef* mul_node1 = graph_def.add_node();
48 NodeDef* add_node2 = graph_def.add_node();
55 NodeDef* add_node3 = graph_def.add_node();
61 NodeDef* const_node1 = graph_def.add_node();
65 NodeDef* const_node2 = graph_def.add_node();
69 NodeDef* const_node3 = graph_def.add_node();
73 NodeDef* add_node4 = graph_def.add_node();
83 TF_ASSERT_OK(RemoveDevice(graph_def, context, &result));
|
| remove_ema_test.cc | 56 GraphDef graph_def; local
57 TF_ASSERT_OK(root.ToGraphDef(&graph_def));
63 FakeQuantizeTraining(graph_def, context, &fake_quantized_graph_def));
66 EXPECT_GT(fake_quantized_graph_def.node_size(), graph_def.node_size());
|
| rename_op_test.cc | 39 GraphDef graph_def; local
41 NodeDef* mul_node1 = graph_def.add_node();
47 NodeDef* add_node2 = graph_def.add_node();
53 NodeDef* add_node3 = graph_def.add_node();
59 NodeDef* const_node1 = graph_def.add_node();
63 NodeDef* const_node2 = graph_def.add_node();
67 NodeDef* const_node3 = graph_def.add_node();
71 NodeDef* add_node4 = graph_def.add_node();
85 TF_ASSERT_OK(RenameOp(graph_def, context, &result));
|
| set_device_test.cc | 38 GraphDef graph_def; local
40 NodeDef* mul_node1 = graph_def.add_node();
47 NodeDef* add_node2 = graph_def.add_node();
54 NodeDef* add_node3 = graph_def.add_node();
60 NodeDef* const_node1 = graph_def.add_node();
64 NodeDef* const_node2 = graph_def.add_node();
68 NodeDef* const_node3 = graph_def.add_node();
72 NodeDef* add_node4 = graph_def.add_node();
78 return graph_def;
83 GraphDef graph_def = CreateDeviceGraph() local
104 GraphDef graph_def = CreateDeviceGraph(); local
[all...] |
| file_utils_test.cc | 41 GraphDef graph_def; local
42 TF_ASSERT_OK(root.ToGraphDef(&graph_def));
46 TF_ASSERT_OK(WriteTextProto(Env::Default(), text_file, graph_def));
50 TF_ASSERT_OK(WriteBinaryProto(Env::Default(), binary_file, graph_def));
59 EXPECT_TRUE(EqualGraphDef(text_graph_def, graph_def, &text_diff))
65 EXPECT_TRUE(EqualGraphDef(binary_graph_def, graph_def, &binary_diff))
|
| obfuscate_names_test.cc | 39 GraphDef graph_def; local
41 NodeDef* add_node1 = graph_def.add_node();
47 NodeDef* add_node2 = graph_def.add_node();
53 NodeDef* add_node3 = graph_def.add_node();
59 NodeDef* const_node1 = graph_def.add_node();
63 NodeDef* const_node2 = graph_def.add_node();
67 NodeDef* const_node3 = graph_def.add_node();
71 NodeDef* const_node4 = graph_def.add_node();
77 ObfuscateNames(graph_def, {{"const_node1"}, {"add_node1"}}, &result));
92 GraphDef graph_def; local
111 GraphDef graph_def; local
[all...] |
| remove_attribute_test.cc | 39 GraphDef graph_def; local
41 NodeDef* mul_node1 = graph_def.add_node();
49 NodeDef* add_node2 = graph_def.add_node();
58 NodeDef* add_node3 = graph_def.add_node();
64 NodeDef* const_node1 = graph_def.add_node();
68 NodeDef* const_node2 = graph_def.add_node();
72 NodeDef* const_node3 = graph_def.add_node();
76 NodeDef* add_node4 = graph_def.add_node();
90 TF_ASSERT_OK(RemoveAttribute(graph_def, context, &wildcard_result));
109 RemoveAttribute(graph_def, targeted_context, &targeted_result))
[all...] |
| rename_attribute_test.cc | 39 GraphDef graph_def; local
41 NodeDef* mul_node1 = graph_def.add_node();
49 NodeDef* add_node2 = graph_def.add_node();
58 NodeDef* add_node3 = graph_def.add_node();
64 NodeDef* const_node1 = graph_def.add_node();
68 NodeDef* const_node2 = graph_def.add_node();
72 NodeDef* const_node3 = graph_def.add_node();
76 NodeDef* add_node4 = graph_def.add_node();
92 TF_ASSERT_OK(RenameAttribute(graph_def, context, &wildcard_result));
115 RenameAttribute(graph_def, targeted_context, &targeted_result))
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| dump_graph.cc | 64 string DumpGraphDefToFile(const string& name, GraphDef const& graph_def) {
66 Status status = WriteTextProto(Env::Default(), path, graph_def);
77 GraphDef graph_def; local
78 graph.ToGraphDef(&graph_def);
80 *graph_def.mutable_library() = flib_def->ToProto();
82 return DumpGraphDefToFile(name, graph_def);
|
| xla_jit_compiled_cpu_function_test.cc | 41 GraphDef graph_def; local
42 NodeDef* x = graph_def.add_node();
46 NodeDef* y = graph_def.add_node();
50 NodeDef* sum = graph_def.add_node();
56 return graph_def;
74 GraphDef graph_def = SumGraph(); local
79 XlaJitCompiledCpuFunction::Compile(graph_def, config,
135 GraphDef graph_def = SumGraph(); local
137 (*graph_def.mutable_node(2)->mutable_attr())["junk"] =
141 EXPECT_FALSE(XlaJitCompiledCpuFunction::Compile(graph_def, config
[all...] |
| tf2xla_test.cc | 41 GraphDef graph_def; local
42 NodeDef* x = graph_def.add_node();
46 NodeDef* y = graph_def.add_node();
50 NodeDef* sum = graph_def.add_node();
56 return graph_def;
68 GraphDef graph_def = SumGraph(); local
73 TF_EXPECT_OK(ConvertGraphDefToXla(graph_def, config, client, &computation));
|
| /external/tensorflow/tensorflow/contrib/util/ |
| convert_graphdef_memmapped_format_test.cc | 29 bool GraphHasImmutableConstNodes(const GraphDef& graph_def) {
30 for (const auto& node : graph_def.node()) {
58 GraphDef graph_def; local
59 TF_ASSERT_OK(root.ToGraphDef(&graph_def));
61 graph_def.SerializeToString(&graph_def_serialized);
109 GraphDef graph_def; local
110 TF_ASSERT_OK(root.ToGraphDef(&graph_def));
112 graph_def.SerializeToString(&graph_def_serialized);
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| hexagon_rewriter_transform_test.cc | 60 GraphDef graph_def; local
61 TF_ASSERT_OK(root.ToGraphDef(&graph_def));
72 RewriteQuantizedStrippedModelForHexagon(graph_def, context, &result));
|
| /external/tensorflow/tensorflow/cc/client/ |
| client_session.cc | 97 GraphDef graph_def; local
98 graph_->ToGraphDefSubRange(&graph_def, last_num_graph_nodes_);
100 return session_->Extend(graph_def);
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| tfcompile_main.cc | 86 GraphDef graph_def; local
87 TF_RETURN_IF_ERROR(ReadProtoFile(flags.graph, &graph_def));
89 TF_RETURN_IF_ERROR(CompileGraph(graph_def, config, flags, &compile_result));
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| graph_to_functiondef_test.cc | 56 GraphDef graph_def; local
57 TF_EXPECT_OK(root.ToGraphDef(&graph_def));
61 TF_EXPECT_OK(ConvertGraphDefToGraph(options, graph_def, graph.get()));
92 GraphDef graph_def; local
93 TF_EXPECT_OK(root.ToGraphDef(&graph_def));
97 TF_EXPECT_OK(ConvertGraphDefToGraph(options, graph_def, graph.get()));
|
| /external/tensorflow/tensorflow/tools/benchmark/ |
| benchmark_model_test.cc | 51 GraphDef graph_def; local
52 TF_ASSERT_OK(root.ToGraphDef(&graph_def));
54 graph_def.SerializeToString(&graph_def_serialized);
|
| /external/tensorflow/tensorflow/cc/tools/ |
| freeze_saved_model.cc | 66 GraphDef* graph_def,
68 for (size_t i = 0; i < graph_def->node_size(); i++) {
69 NodeDef* node = graph_def->mutable_node(i);
77 GraphDef* graph_def, const std::unordered_set<string>& outputs,
87 GetNodeNameToNodeDefMap(graph_def, &name_to_node_map);
150 GraphDef graph_def = saved_model_bundle.meta_graph_def.graph_def(); local
152 *frozen_graph_def->mutable_versions() = graph_def.versions();
153 *frozen_graph_def->mutable_library() = graph_def.library();
155 if (graph_def.node_size() == 0)
[all...] |
| /external/tensorflow/tensorflow/contrib/hvx/hvx_ops_support_checker/ |
| hvx_ops_support_checker_main.cc | 19 // --in_graph=graph_def.pb
111 static void CheckOpsSupport(const GraphDef& graph_def,
116 LOG(INFO) << "Checking " << graph_def.node_size() << " nodes";
123 for (const NodeDef& node : graph_def.node()) {
152 for (const NodeDef& node : graph_def.node()) {
171 tensorflow::GraphDef graph_def; local
173 in_graph, &graph_def));
175 tensorflow::CheckOpsSupport(graph_def, dump_all_nodes, dump_shape_and_type);
|
