| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| transform_graph.h | 45 GraphDef* graph_def);
|
| transform_graph_test.cc | 38 Status test_empty_graph_transform(const GraphDef& graph_def,
75 GraphDef graph_def; local
76 TF_ASSERT_OK(root.ToGraphDef(&graph_def));
78 graph_def.SerializeToString(&graph_def_serialized);
131 GraphDef graph_def; local
132 TF_ASSERT_OK(root.ToGraphDef(&graph_def));
133 EXPECT_EQ(1, graph_def.node().size());
135 &graph_def));
136 EXPECT_EQ(0, graph_def.node().size());
138 TF_ASSERT_OK(root.ToGraphDef(&graph_def));
[all...] |
| transform_graph.cc | 197 GraphDef graph_def; local
198 Status load_status = LoadTextOrBinaryGraphFile(in_graph, &graph_def);
207 TransformGraph(inputs, outputs, transform_params, &graph_def);
217 save_status = WriteTextProto(Env::Default(), out_graph, graph_def);
219 save_status = WriteBinaryProto(Env::Default(), out_graph, graph_def);
253 GraphDef* graph_def) {
276 transform_func(*graph_def, context, &transformed_graph_def);
281 transformed_graph_def = *graph_def;
287 *transformed_graph_def.mutable_library() = graph_def->library();
290 *graph_def = transformed_graph_def
[all...] |
| 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());
|
| /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/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/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/contrib/receptive_field/python/util/examples/ |
| compute_rf.py | 45 graph_def: A GraphDef object.
47 graph_def = graph_pb2.GraphDef()
49 text_format.Parse(pbstr, graph_def)
50 return graph_def
55 graph_def = _load_graphdef(cmd_args.graph_path)
60 graph_def, cmd_args.input_node, cmd_args.output_node)
|
| /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);
|
| /external/tensorflow/tensorflow/contrib/lite/toco/tensorflow_graph_matching/ |
| resolve_cluster.cc | 64 const GraphDef& graph_def,
67 for (const NodeDef& node : graph_def.node()) {
73 cluster_factory.CreateCluster(node, graph_def);
88 const GraphDef& graph_def,
94 for (const NodeDef& node : graph_def.node()) {
103 if (FindCluster(*cluster_factory, graph_def, &is_node_in_cluster,
122 for (const NodeDef& node : graph_def.node()) {
|
| resolve_svdf.h | 77 const tensorflow::GraphDef& graph_def) const;
|
| /external/tensorflow/tensorflow/contrib/tensorrt/segment/ |
| segment_test.cc | 32 bool GetGraphDef(TF_Graph* graph, tensorflow::GraphDef* graph_def);
51 tensorflow::GraphDef* graph_def) {
57 if (ret) ret = graph_def->ParseFromArray(buffer->data, buffer->length);
110 GraphDef graph_def; local
111 ASSERT_TRUE(GetGraphDef(graph, &graph_def));
115 SegmentGraph(graph_def, MakeCandidateFn({}), default_options_, &segments),
154 GraphDef graph_def; local
155 ASSERT_TRUE(GetGraphDef(graph, &graph_def));
159 SegmentGraph(graph_def,
209 GraphDef graph_def; local
266 GraphDef graph_def; local
336 GraphDef graph_def; local
[all...] |
| /external/tensorflow/tensorflow/python/framework/ |
| meta_graph.py | 107 filename: `graph_def` filename including the path.
115 graph_def = graph_pb2.GraphDef()
121 graph_def.ParseFromString(file_content)
122 return graph_def
128 text_format.Merge(file_content, graph_def)
132 return graph_def
135 def ops_used_by_graph_def(graph_def):
141 graph_def: A `GraphDef` proto, as from `graph.as_graph_def()`.
148 for fun in graph_def.library.function:
161 for node in graph_def.node
[all...] |
| graph_util_test.py | 164 graph_def = graph_pb2.GraphDef()
165 n1 = graph_def.node.add()
168 n2 = graph_def.node.add()
172 n3 = graph_def.node.add()
177 n4 = graph_def.node.add()
181 n5 = graph_def.node.add()
185 sub_graph = graph_util.extract_sub_graph(graph_def, ["n3"])
192 graph_def = graph_pb2.GraphDef()
193 n1 = graph_def.node.add()
196 graph_util.extract_sub_graph(graph_def, "n1"
[all...] |
| /external/tensorflow/tensorflow/contrib/meta_graph_transform/ |
| meta_graph_transform_test.py | 72 expected_meta_graph_def.graph_def.ClearField('versions')
74 expected_meta_graph_def.graph_def.library.CopyFrom(
104 graph_def = graph_pb2.GraphDef()
105 graph_def.node.extend([node_def_pb2.NodeDef(name='z1', op='NoOp')])
116 graph_def, input_names, output_names, initializer_names, transforms))
120 graph_def, input_names, output_names + init_nodes, transforms)
124 graph_def = graph_pb2.GraphDef()
125 graph_def.node.extend([node_def_pb2.NodeDef(name='z1', op='NoOp')])
137 graph_def.node.extend([node_def_pb2.NodeDef(name='n1', op='NoOp')])
143 meta_graph_transform._freeze_transform(graph_def, output_names
[all...] |
| /external/tensorflow/tensorflow/contrib/framework/python/framework/ |
| graph_util.py | 37 def fuse_op(graph_def, input_nodes, output_nodes, output_dtypes,
42 graph_def: A graph_pb2.GraphDef proto.
53 TypeError: If 'graph_def' is not a graph_pb2.GraphDef proto.
56 if not isinstance(graph_def, graph_pb2.GraphDef):
57 raise TypeError("graph_def must be a graph_pb2.GraphDef proto.")
66 graph_def)
82 for node in graph_def.node:
128 out.library.CopyFrom(graph_def.library)
129 out.versions.CopyFrom(graph_def.versions)
|
| graph_util_test.py | 42 graph_def = graph_pb2.GraphDef()
48 graph_def.node.extend([node_a, node_b, node_c, node_d, node_e])
50 graph_def, ['A'], ['D'], [types_pb2.DT_FLOAT], True, 'FusedOp', 'Op2')
63 graph_def = graph_pb2.GraphDef()
70 graph_def.node.extend([node_a, node_a1, node_b, node_c, node_d, node_e])
71 fused_graph_def = graph_util.fuse_op(graph_def, ['A', 'A1'], ['D'],
|
| /external/tensorflow/tensorflow/contrib/receptive_field/python/util/ |
| receptive_field_test.py | 288 graph_def = create_test_network_1().as_graph_def()
294 graph_def, input_node, output_node))
303 graph_def = create_test_network_2().as_graph_def()
309 graph_def, input_node, output_node))
318 graph_def = create_test_network_3().as_graph_def()
323 graph_def, input_node, output_node)
326 graph_def = create_test_network_4().as_graph_def()
332 graph_def, input_node, output_node))
341 graph_def = create_test_network_4().as_graph_def()
347 graph_def, input_node, output_node, input_resolution=[9, 9])
[all...] |
| /external/tensorflow/tensorflow/python/grappler/ |
| cost_analyzer_tool.py | 52 graph_def = graph_pb2.GraphDef()
54 text_format.Merge(graph_file.read(), graph_def)
56 graph_def.ParseFromString(graph_file.read())
57 importer.import_graph_def(graph_def, name="")
62 graph_def=graph.as_graph_def(), graph=graph)
68 metagraph.graph_def.CopyFrom(optimized_graph)
|
| /external/tensorflow/tensorflow/core/kernels/fuzzing/ |
| fuzz_session.h | 97 GraphDef graph_def; local
98 TF_CHECK_OK(root.ToGraphDef(&graph_def));
100 Status status = session_->Create(graph_def);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| remote_fused_graph_execute_utils.h | 97 const GraphDef& graph_def,
108 const GraphDef& graph_def,
136 static bool GetOutputTensorShapeType(const GraphDef& graph_def,
141 const GraphDef& graph_def,
163 const bool dry_run_inference, GraphDef* graph_def);
189 const GraphDef& graph_def,
194 const GraphDef& graph_def,
201 const GraphDef& graph_def,
255 const string& remote_graph_executor_name, GraphDef* graph_def);
283 const GraphDef& graph_def, const std::unordered_set<string>& op_types)
[all...] |
| immutable_constant_op_test.cc | 98 GraphDef graph_def; local
99 TF_ASSERT_OK(root.ToGraphDef(&graph_def));
107 TF_ASSERT_OK(session->Create(graph_def)) << "Can't create test graph";
130 GraphDef graph_def; local
131 TF_ASSERT_OK(root.ToGraphDef(&graph_def));
136 TF_ASSERT_OK(session->Create(graph_def)) << "Can't create test graph";
171 GraphDef graph_def; local
172 TF_ASSERT_OK(root.ToGraphDef(&graph_def));
179 TF_ASSERT_OK(session->Create(graph_def)) << "Can't create test graph";
|
| /external/tensorflow/tensorflow/python/tools/ |
| import_pb_to_tensorboard.py | 47 graph_def = graph_pb2.GraphDef()
48 graph_def.ParseFromString(f.read())
49 importer.import_graph_def(graph_def)
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| tf2xla_util.h | 32 // Modifies <graph_def> to include placeholders for each fed tensor, and
39 std::unordered_map<string, string>* feed_remapping, GraphDef* graph_def);
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| master.proto | 38 GraphDef graph_def = 1;
63 // The "graph_def" specifies a set of nodes to be added to the session's graph.
65 // A typical "graph_def" will contain:
71 // None of the names in req.graph_def appeared in previous successful calls to
85 GraphDef graph_def = 2;
|
