| /external/tensorflow/tensorflow/python/framework/ |
| meta_graph_test.py | 91 graph_def=ops.get_default_graph().as_graph_def(add_shapes=True),
180 graph_def=ops.get_default_graph().as_graph_def(),
183 meta_graph_def.graph_def)
190 graph_def=ops.get_default_graph().as_graph_def(),
193 meta_graph_def.graph_def)
206 graph_def=ops.get_default_graph().as_graph_def(),
209 meta_graph_def.graph_def)
227 graph_def=ops.get_default_graph().as_graph_def(),
231 for function_def in meta_graph_def.graph_def.library.function:
246 graph_def = graph_pb2.GraphDef(
[all...] |
| test_util.py | 170 assert_equal_graph_def(a.graph_def, b.graph_def, checkpoint_v2=True)
171 # Check graph_def versions (ignored by assert_equal_graph_def).
172 tester.assertProtoEquals(a.graph_def.versions, b.graph_def.versions)
175 a.ClearField("graph_def")
176 b.ClearField("graph_def")
186 def _strip_checkpoint_v2_randomized(graph_def):
187 for node in graph_def.node:
[all...] |
| /external/tensorflow/tensorflow/core/graph/ |
| graph_partition_test.cc | 75 void Partition(const GraphDef& graph_def,
79 TF_CHECK_OK(ConvertGraphDefToGraph(opts, graph_def, &g));
100 EXPECT_EQ(graph_def.versions().producer(), TF_GRAPH_DEF_VERSION);
103 EXPECT_EQ(graph_def.versions().producer(), it.second.versions().producer());
104 EXPECT_EQ(graph_def.versions().min_consumer(),
109 void CheckLoopConstruction(const GraphDef& graph_def) {
111 Partition(graph_def, &partitions);
210 GraphDef graph_def; local
211 TF_EXPECT_OK(scope_a_.ToGraphDef(&graph_def));
213 TF_EXPECT_GRAPH_EQ(graph_def, partitions_[a])
217 GraphDef graph_def; local
[all...] |
| graph_def_builder.cc | 67 Status GraphDefBuilder::ToGraphDef(GraphDef* graph_def) const {
69 graph_.ToGraphDef(graph_def);
|
| optimizer_cse_test.cc | 38 GraphDef graph_def; local
42 CHECK(parser.MergeFromString(s, &graph_def)) << s;
44 TF_CHECK_OK(ConvertGraphDefToGraph(opts, graph_def, graph));
|
| graph_def_builder.h | 57 // GraphDef graph_def;
58 // Status status = b.ToGraphDef(&graph_def);
64 // b.ToGraphDef(&graph_def);
161 // successful, and if so fill *graph_def.
162 Status ToGraphDef(GraphDef* graph_def) const;
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| transform_utils.cc | 63 void MapNamesToNodes(const GraphDef& graph_def,
65 for (const NodeDef& node : graph_def.node()) {
70 void MapNodesToOutputs(const GraphDef& graph_def,
73 MapNamesToNodes(graph_def, &node_map);
74 for (const NodeDef& node : graph_def.node()) {
278 GraphMatcher::GraphMatcher(const GraphDef& graph_def) {
279 SortByExecutionOrder(graph_def, &graph_def_).IgnoreError();
549 void FindInvalidInputs(const GraphDef& graph_def,
552 MapNamesToNodes(graph_def, &node_map);
554 for (const NodeDef& node : graph_def.node())
[all...] |
| /external/tensorflow/tensorflow/core/debug/ |
| debug_grpc_testlib.cc | 41 } else if (!event.graph_def().empty()) {
42 encoded_graph_defs.push_back(event.graph_def());
|
| /external/tensorflow/tensorflow/contrib/session_bundle/ |
| exporter.py | 155 graph_def=None,
165 graph_def: A GraphDef message of the graph to be used in inference.
198 if graph_def or clear_devices:
200 if graph_def:
201 copy.CopyFrom(graph_def)
|
| exporter_test.py | 145 # Validate custom graph_def.
148 graph_def = graph_pb2.GraphDef()
149 graph_def_any[0].Unpack(graph_def)
153 self.assertProtoEquals(compare_def, graph_def)
|
| session_bundle_test.cc | 72 MetaGraphDef graph_def; local
73 TF_RETURN_IF_ERROR(ReadBinaryProto(Env::Default(), source, &graph_def));
74 twiddler(&graph_def);
76 WriteStringToFile(Env::Default(), sink, graph_def.SerializeAsString()));
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| message_wrappers_test.cc | 108 GraphDef graph_def; local
109 graph_def.mutable_versions()->set_producer(1234);
110 graph_def.mutable_versions()->set_min_consumer(1234);
111 run_graph_response->AddPartitionGraph(graph_def);
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| single_machine.cc | 131 Status SingleMachine::Run(const GraphDef& graph_def,
137 if (last_graph_ != &graph_def) {
139 TF_RETURN_IF_ERROR(session_->Create(graph_def));
178 last_graph_ = &graph_def;
|
| /external/tensorflow/tensorflow/examples/speech_commands/ |
| label_wav.cc | 40 tensorflow::GraphDef graph_def; local
42 ReadBinaryProto(tensorflow::Env::Default(), graph_file_name, &graph_def);
48 Status session_create_status = (*session)->Create(graph_def);
|
| test_streaming_accuracy.cc | 100 tensorflow::GraphDef graph_def; local
102 ReadBinaryProto(tensorflow::Env::Default(), graph_file_name, &graph_def);
108 Status session_create_status = (*session)->Create(graph_def);
|
| /external/tensorflow/tensorflow/python/grappler/ |
| item_test.py | 74 for node in grappler_item.metagraph.graph_def.node:
99 for node in grappler_item.metagraph.graph_def.node:
105 for node in grappler_item.metagraph.graph_def.node:
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| tf2xla.cc | 305 // InitGraph creates a graph based on the graph_def, that may then be converted
315 Status InitGraph(const GraphDef& graph_def, const tf2xla::Config& config,
319 FunctionLibraryDefinition flib_def(OpRegistry::Global(), graph_def.library());
324 GraphDef first_copy_def = graph_def;
349 Status ConvertGraphDefToXla(const GraphDef& graph_def,
353 TF_RETURN_IF_ERROR(InitGraph(graph_def, config, &graph));
|
| xla_jit_compiled_cpu_function.cc | 113 const GraphDef& graph_def, const tf2xla::Config& config,
115 // Convert the graph_def into an xla::Computation.
119 TF_RETURN_IF_ERROR(tensorflow::ConvertGraphDefToXla(graph_def, config, client,
|
| /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/c/ |
| c_test_util.cc | 340 bool GetGraphDef(TF_Graph* graph, tensorflow::GraphDef* graph_def) {
346 if (ret) ret = graph_def->ParseFromArray(buffer->data, buffer->length);
387 const tensorflow::GraphDef& graph_def) {
389 for (const tensorflow::GradientDef& grad : graph_def.library().gradient()) {
396 std::vector<string> GetFuncNames(const tensorflow::GraphDef& graph_def) {
398 for (const tensorflow::FunctionDef& func : graph_def.library().function()) {
|
| /external/tensorflow/tensorflow/core/grappler/ |
| grappler_item_builder.cc | 88 GraphDef graph_def(graph_def_arg);
93 for (auto& func : *graph_def.mutable_library()->mutable_function()) {
104 graph_def.library());
120 graph_def.versions().producer(),
131 ConvertGraphDefToGraph(graph_ctor_opts, graph_def, graphptr.get()));
166 new_item->graph = meta_graph.graph_def();
|
| /external/tensorflow/tensorflow/core/framework/ |
| dataset.cc | 206 GraphDef graph_def; local
207 TF_RETURN_IF_ERROR(b.ToGraphDef(&graph_def));
208 graph_def.SerializeToString(serialized_graph_def);
|
| /external/tensorflow/tensorflow/examples/label_image/ |
| main.cc | 187 tensorflow::GraphDef graph_def; local
189 ReadBinaryProto(tensorflow::Env::Default(), graph_file_name, &graph_def);
195 Status session_create_status = (*session)->Create(graph_def);
|
| /external/tensorflow/tensorflow/python/tools/ |
| freeze_graph.py | 94 for node in input_meta_graph_def.graph_def.node:
145 input_meta_graph_def.graph_def,
231 input_saved_model_dir, saved_model_tags).graph_def
|
| /external/tensorflow/tensorflow/core/grappler/inputs/ |
| file_input_yielder.cc | 100 for (auto& node : metagraph.graph_def().node()) {
|
