| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| sort_by_execution_order_test.cc | 33 void GetOrder(const GraphDef& graph_def, std::map<string, int>* order) {
34 for (int i = 0; i < graph_def.node_size(); ++i) {
35 const NodeDef& node = graph_def.node(i);
41 GraphDef graph_def; local
42 NodeDef* add_node = graph_def.add_node();
48 NodeDef* b_node = graph_def.add_node();
52 NodeDef* a_node = graph_def.add_node();
57 TF_ASSERT_OK(SortByExecutionOrder(graph_def, &result));
67 GraphDef graph_def; local
69 NodeDef* negative_node = graph_def.add_node()
100 GraphDef graph_def; local
151 GraphDef graph_def; local
[all...] |
| strip_unused_nodes_test.cc | 40 GraphDef graph_def; local
41 NodeDef* add_node = graph_def.add_node();
47 NodeDef* a_node = graph_def.add_node();
51 NodeDef* b_node = graph_def.add_node();
55 NodeDef* c_node = graph_def.add_node();
60 TF_ASSERT_OK(StripUnusedNodes(graph_def, {{}, {"add_node"}}, &result));
71 GraphDef graph_def; local
73 NodeDef* add_node1 = graph_def.add_node();
79 NodeDef* add_node2 = graph_def.add_node();
85 NodeDef* add_node3 = graph_def.add_node()
131 GraphDef graph_def; local
174 GraphDef graph_def; local
220 GraphDef graph_def; local
[all...] |
| remove_nodes_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* identity_node1 = graph_def.add_node();
64 NodeDef* identity_node2 = graph_def.add_node();
69 NodeDef* identity_node3 = graph_def.add_node();
74 NodeDef* const_node1 = graph_def.add_node();
78 NodeDef* const_node2 = graph_def.add_node();
82 NodeDef* const_node3 = graph_def.add_node()
126 GraphDef graph_def; local
165 GraphDef graph_def; local
215 GraphDef graph_def; local
[all...] |
| backports_test.cc | 42 GraphDef graph_def; local
44 NodeDef* mul_node1 = graph_def.add_node();
50 NodeDef* add_node2 = graph_def.add_node();
56 NodeDef* add_node3 = graph_def.add_node();
62 NodeDef* const_node1 = graph_def.add_node();
66 NodeDef* const_node2 = graph_def.add_node();
70 NodeDef* const_node3 = graph_def.add_node();
74 NodeDef* concat_node = graph_def.add_node();
86 TF_ASSERT_OK(BackportConcatV2Transform(graph_def, context, &result));
108 GraphDef graph_def; local
180 GraphDef graph_def; local
[all...] |
| 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...] |
| sparsify_gather_test.cc | 42 const std::vector<NodeDef*>& inputs, GraphDef* graph_def,
44 NodeDef* node_def = graph_def->add_node();
60 NodeDef* indices, GraphDef* graph_def) {
63 CreateNode(strings::StrCat(name, "_axis"), "Const", {}, graph_def);
67 CreateNode(name, "GatherV2", {params, indices, axis_node}, graph_def);
69 CreateNode(name, "Gather", {params, indices}, graph_def);
76 GraphDef graph_def; local
81 NodeDef* input_node = CreateNode("ids", "Const", {}, &graph_def);
92 w_node = CreateNode("w/part_1", "Const", {}, &graph_def);
95 w_node = CreateNode("w/part_1", "VariableV2", {}, &graph_def);
289 GraphDef graph_def; local
[all...] |
| 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));
|
| 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...] |
| insert_logging_test.cc | 38 void CheckGraphCanRun(const GraphDef& graph_def,
41 TF_ASSERT_OK(session->Create(graph_def));
64 GraphDef graph_def; local
65 TF_ASSERT_OK(root.ToGraphDef(&graph_def));
66 CheckGraphCanRun(graph_def, {"add_node4"});
72 TF_ASSERT_OK(InsertLogging(graph_def, context, &result));
111 GraphDef graph_def; local
112 TF_ASSERT_OK(root.ToGraphDef(&graph_def));
113 CheckGraphCanRun(graph_def, {"add_node4"});
121 TF_ASSERT_OK(InsertLogging(graph_def, context, &result))
160 GraphDef graph_def; local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_tfgraph_builder_test.cc | 73 GraphDef graph_def = generator_.GetGraphDef(); local
74 EXPECT_EQ(graph_def.node_size(), 3);
75 const auto &node = graph_def.node(2);
100 GraphDef graph_def = generator_.GetGraphDef(); local
101 EXPECT_EQ(graph_def.node_size(), 1);
102 const auto &node = graph_def.node(0);
112 GraphDef graph_def = generator_.GetGraphDef(); local
113 EXPECT_EQ(graph_def.node_size(), 2);
114 EXPECT_EQ(graph_def.node(0).name(), "Negate/param0.0");
115 EXPECT_EQ(graph_def.node(0).op(), "HloParameter")
131 GraphDef graph_def = generator_.GetGraphDef(); local
153 GraphDef graph_def = generator_.GetGraphDef(); local
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| graph_def_util.h | 30 string SummarizeGraphDef(const GraphDef& graph_def);
44 Status ValidateExternalGraphDefSyntax(const GraphDef& graph_def);
46 // Adds default attributes to NodeDefs in 'graph_def' starting
47 // from the 'node_offset' node in 'graph_def'.
51 // Returns OK on success, an error if 'graph_def' has a NodeDef
54 // REQUIRES: 'graph_def' and 'op_registry' are not nullptr.
55 Status AddDefaultAttrsToGraphDef(GraphDef* graph_def,
59 // Remove attrs from 'graph_def' that have the default value according
61 // 'consumer_op_registry'. This can allow 'graph_def' to run on the
71 // // For each consumer, update 'graph_def'
[all...] |
| graph_def_util.cc | 36 string SummarizeGraphDef(const GraphDef& graph_def) {
39 &ret, "versions = ", ProtoShortDebugString(graph_def.versions()), ";\n");
40 for (const NodeDef& node : graph_def.node()) {
46 Status ValidateExternalGraphDefSyntax(const GraphDef& graph_def) {
47 for (const NodeDef& node : graph_def.node()) {
53 Status AddDefaultAttrsToGraphDef(GraphDef* graph_def,
56 if (node_offset > graph_def->node_size()) {
60 node_offset, " with total nodes in graph: ", graph_def->node_size());
63 for (int i = node_offset; i < graph_def->node_size(); ++i) {
64 NodeDef* node_def = graph_def->mutable_node(i)
[all...] |
| graph_def_util_test.cc | 46 GraphDef graph_def; local
48 .Finalize(graph_def.add_node()));
49 GraphDef expected_graph_def = graph_def;
52 TF_ASSERT_OK(RemoveNewDefaultAttrsFromGraphDef(&graph_def, registry, registry,
55 TF_EXPECT_GRAPH_EQ(expected_graph_def, graph_def);
66 GraphDef graph_def; local
69 .Finalize(graph_def.add_node()));
70 GraphDef expected_graph_def = graph_def;
73 TF_ASSERT_OK(RemoveNewDefaultAttrsFromGraphDef(&graph_def, registry, registry,
76 TF_EXPECT_GRAPH_EQ(expected_graph_def, graph_def);
244 GraphDef graph_def; local
291 GraphDef graph_def; local
[all...] |
| /external/tensorflow/tensorflow/core/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...] |
| validate.cc | 29 Status ValidateGraphDef(const GraphDef& graph_def,
32 const int version = graph_def.versions().producer();
33 for (const NodeDef& node_def : graph_def.node()) {
45 const GraphDef& graph_def, const OpRegistryInterface& op_registry) {
46 GraphDef copy(graph_def);
51 Status ValidateGraphDefAgainstOpList(const GraphDef& graph_def,
54 return ValidateGraphDefAgainstOpRegistry(graph_def, registry);
|
| validate.h | 26 // Returns OK if every NodeDef in `graph_def` is valid with respect to
32 // * `graph_def` has default attrs filled in (see AddDefaultAttrsToGraphDef()).
33 Status ValidateGraphDef(const GraphDef& graph_def,
36 // Like ValidateGraphDef() except it makes a copy of `graph_def` and calls
40 const GraphDef& graph_def, const OpRegistryInterface& op_registry);
46 Status ValidateGraphDefAgainstOpList(const GraphDef& graph_def,
|
| default_device.h | 27 // Sets the default device for all nodes in graph_def to "device",
29 inline void SetDefaultDevice(const string& device, GraphDef* graph_def) {
30 for (int i = 0; i < graph_def->node_size(); ++i) {
31 auto node = graph_def->mutable_node(i);
|
| 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);
|
| /external/tensorflow/tensorflow/cc/tools/ |
| freeze_saved_model_test.cc | 58 // Adds an initialized session to `saved_model_bundle` using `graph_def` and
61 const GraphDef& graph_def, const string& init_node,
65 TF_RETURN_IF_ERROR(saved_model_bundle->session->Create(graph_def));
74 // Adds `graph_def` to `saved_model_bundle` and initializes a session with
76 Status AddGraphDefToSavedModelBundle(const GraphDef& graph_def,
80 *meta_graph_def->mutable_graph_def() = graph_def;
81 return InitializeSavedModelBundleSession(graph_def, init_node,
85 // Adds `graph_def` and `outputs` as the GraphDef and SignatureDef in
88 const GraphDef& graph_def, const std::unordered_set<string>& outputs,
94 return AddGraphDefToSavedModelBundle(graph_def, init_node
162 GraphDef graph_def; local
181 GraphDef graph_def; local
204 GraphDef graph_def; local
240 GraphDef graph_def; local
274 GraphDef graph_def; local
[all...] |
| /external/tensorflow/tensorflow/cc/training/ |
| queue_runner_test.cc | 78 GraphDef graph_def; local
79 TF_EXPECT_OK(root.ToGraphDef(&graph_def));
80 return graph_def;
102 const GraphDef& graph_def) {
105 TF_CHECK_OK(session->Create(graph_def));
112 GraphDef graph_def = BuildSimpleGraph(); local
113 auto session = BuildSessionAndInitVariable(graph_def);
130 GraphDef graph_def = BuildSimpleGraph(); local
131 auto session = BuildSessionAndInitVariable(graph_def);
150 GraphDef graph_def = BuildSimpleGraph() local
165 GraphDef graph_def = BuildSimpleGraph(); local
196 GraphDef graph_def; local
202 auto graph_def = BuildDoubleQueueGraph(); local
239 auto graph_def = BuildDoubleQueueGraph(); local
284 GraphDef graph_def = BuildDoubleQueueGraph(); local
301 auto graph_def = BuildDoubleQueueGraph(); local
333 GraphDef graph_def = BuildSimpleGraph(); local
360 GraphDef graph_def; local
394 GraphDef graph_def = BuildSimpleGraph(); local
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| 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/python/summary/writer/ |
| writer.py | 48 def __init__(self, event_writer, graph=None, graph_def=None):
75 graph_def: DEPRECATED: Use the `graph` argument instead.
80 if graph is not None or graph_def is not None:
81 # Calling it with both graph and graph_def for backward compatibility.
82 self.add_graph(graph=graph, graph_def=graph_def)
84 # graph may itself be a graph_def due to positional arguments
88 meta_graph.create_meta_graph_def(graph_def=graph_def or
156 def _add_graph_def(self, graph_def, global_step=None)
[all...] |
| /external/tensorflow/tensorflow/python/framework/ |
| importer.py | 140 return 'graph_def is invalid at node %r: %s.' % (node.name, message)
153 def _ProcessGraphDefParam(graph_def, op_dict):
154 """Type-checks and possibly canonicalizes `graph_def`."""
155 if not isinstance(graph_def, graph_pb2.GraphDef):
156 # `graph_def` could be a dynamically-created message, so try a duck-typed
159 old_graph_def = graph_def
160 graph_def = graph_pb2.GraphDef()
161 graph_def.MergeFrom(old_graph_def)
163 raise TypeError('graph_def must be a GraphDef proto.')
165 # If we're using the graph_def provided by the caller, modify graph_de
[all...] |
