| /external/tensorflow/tensorflow/core/graph/ |
| 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);
|
| 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);
|
| 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,
|
| 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);
|
| 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(),
|
| 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/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...] |
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| file_utils.h | 27 Status LoadTextOrBinaryGraphFile(const string& file_name, GraphDef* graph_def);
|
| 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));
|
| 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...] |
| 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...] |
| file_utils.cc | 23 Status LoadTextOrBinaryGraphFile(const string& file_name, GraphDef* graph_def) {
32 Status load_status = ReadBinaryProto(Env::Default(), file_name, graph_def);
34 if (protobuf::TextFormat::ParseFromString(file_data, graph_def)) {
|
| 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...] |
| 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());
|
| 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))
|
| 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...] |
| 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...] |
| 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...] |
| 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));
|
| /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/compiler/tf2xla/ |
| tf2xla.h | 33 Status ConvertGraphDefToXla(const GraphDef& graph_def,
|
| 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));
|
| 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);
|
