| /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 | 44 GraphDef graph_def; local
46 CHECK(parser.MergeFromString(graph_def_str, &graph_def)) << graph_def_str;
47 TF_ASSERT_OK(graph::ValidateGraphDef(graph_def, *OpRegistry::Global()));
59 GraphDef graph_def; local
61 CHECK(parser.MergeFromString(graph_def_str, &graph_def)) << graph_def_str;
62 Status s = graph::ValidateGraphDef(graph_def, *OpRegistry::Global());
67 TF_ASSERT_OK(AddDefaultAttrsToGraphDef(&graph_def, *OpRegistry::Global(), 0));
70 TF_ASSERT_OK(graph::ValidateGraphDef(graph_def, *OpRegistry::Global()));
82 GraphDef graph_def; local
84 CHECK(parser.MergeFromString(graph_def_str, &graph_def)) << graph_def_str
104 GraphDef graph_def; local
116 GraphDef graph_def; local
159 GraphDef graph_def; local
174 GraphDef graph_def; local
[all...] |
| benchmark_testlib.h | 74 GraphDef graph_def; local
86 *graph_def.add_node() = std::move(node);
97 *graph_def.add_node() = std::move(node);
109 *graph_def.add_node() = std::move(node);
111 return graph_def;
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| dataset_ops.cc | 41 GraphDef graph_def; variable
42 OP_REQUIRES_OK(ctx, b.ToGraphDef(&graph_def));
45 result->scalar<string>()() = graph_def.SerializeAsString();
|
| /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));
73 StatSummarizer stats(graph_def);
|
| dump_graph.cc | 112 string DumpGraphDefToFile(const string& name, GraphDef const& graph_def,
114 return WriteTextProtoToUniqueFile(Env::Default(), name, "GraphDef", graph_def,
121 GraphDef graph_def; local
122 graph.ToGraphDef(&graph_def);
124 *graph_def.mutable_library() = flib_def->ToProto();
126 return DumpGraphDefToFile(name, graph_def, dirname);
|
| /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));
|
| 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...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| 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...] |
| /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/common_runtime/data/ |
| standalone_test.cc | 153 GraphDef graph_def; local
154 protobuf::TextFormat::ParseFromString(kGraphProto, &graph_def);
165 auto s = Dataset::FromGraph({}, graph_def, test_case.fetch_node, &dataset);
|
| /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/core/lib/strings/ |
| proto_serialization_test.cc | 31 GraphDef graph_def; local
33 NodeDef* node = graph_def.add_node();
39 return graph_def;
45 GraphDef graph_def = MakeGraphDef(num_nodes); local
50 SerializeToStringDeterministic(graph_def, &serialized));
58 GraphDef graph_def = MakeGraphDef(num_nodes); local
60 const size_t size = graph_def.ByteSizeLong();
64 SerializeToBufferDeterministic(graph_def, buf.data(), size));
72 GraphDef graph_def = MakeGraphDef(num_nodes); local
75 testing::DoNotOptimize(DeterministicProtoHash64(graph_def));
[all...] |
| /external/tensorflow/tensorflow/lite/tools/accuracy/ |
| eval_pipeline_builder.cc | 88 GraphDef graph_def; local
89 TF_RETURN_IF_ERROR(scope.ToGraphDef(&graph_def));
94 absl::make_unique<EvalPipeline>(graph_def, params, accuracy_eval_);
|
| eval_pipeline_test.cc | 60 GraphDef graph_def; local
61 TF_CHECK_OK(scope.ToGraphDef(&graph_def));
67 EvalPipeline eval_pipeline(graph_def, params, &accuracy_eval);
85 GraphDef graph_def; local
86 TF_CHECK_OK(scope.ToGraphDef(&graph_def));
92 EvalPipeline eval_pipeline(graph_def, params, &accuracy_eval);
108 GraphDef graph_def; local
109 TF_CHECK_OK(scope.ToGraphDef(&graph_def));
115 EvalPipeline eval_pipeline(graph_def, params, &accuracy_eval);
|
| file_reader_stage_test.cc | 69 GraphDef graph_def; local
70 TF_CHECK_OK(scope.ToGraphDef(&graph_def));
72 TF_CHECK_OK(session->Create(graph_def));
89 GraphDef graph_def; local
90 TF_CHECK_OK(scope.ToGraphDef(&graph_def));
92 TF_CHECK_OK(session->Create(graph_def));
|
| run_tflite_model_op_test.cc | 69 GraphDef graph_def; local
70 TF_CHECK_OK(scope.ToGraphDef(&graph_def));
73 TF_CHECK_OK(session->Create(graph_def));
129 GraphDef graph_def; local
130 TF_CHECK_OK(scope.ToGraphDef(&graph_def));
132 TF_CHECK_OK(session->Create(graph_def));
176 GraphDef graph_def; local
177 TF_CHECK_OK(scope.ToGraphDef(&graph_def));
179 TF_CHECK_OK(session->Create(graph_def));
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| tfcompile_main.cc | 87 GraphDef graph_def; local
88 TF_RETURN_IF_ERROR(ReadProtoFile(flags.graph, &graph_def));
90 TF_RETURN_IF_ERROR(CompileGraph(graph_def, config, flags, &compile_result));
|
| /external/tensorflow/tensorflow/core/framework/ |
| graph_to_functiondef_test.cc | 65 GraphDef graph_def; local
66 TF_EXPECT_OK(root.ToGraphDef(&graph_def));
70 TF_EXPECT_OK(ConvertGraphDefToGraph(options, graph_def, graph.get()));
102 GraphDef graph_def; local
103 TF_EXPECT_OK(root.ToGraphDef(&graph_def));
107 TF_EXPECT_OK(ConvertGraphDefToGraph(options, graph_def, graph.get()));
|
| /external/tensorflow/tensorflow/core/kernels/fuzzing/ |
| fuzz_session.h | 96 GraphDef graph_def; local
97 TF_CHECK_OK(root.ToGraphDef(&graph_def));
99 Status status = session_->Create(graph_def);
|
| /external/tensorflow/tensorflow/lite/tools/accuracy/ilsvrc/ |
| inception_preprocessing_test.cc | 65 GraphDef graph_def; local
66 TF_CHECK_OK(scope.ToGraphDef(&graph_def));
68 TF_CHECK_OK(session->Create(graph_def));
94 GraphDef graph_def; local
95 TF_CHECK_OK(scope.ToGraphDef(&graph_def));
97 TF_CHECK_OK(session->Create(graph_def));
123 GraphDef graph_def; local
124 TF_CHECK_OK(scope.ToGraphDef(&graph_def));
126 TF_CHECK_OK(session->Create(graph_def));
|
| /external/tensorflow/tensorflow/tools/benchmark/ |
| benchmark_model_test.cc | 31 string* output_name, GraphDef* graph_def) {
47 TF_ASSERT_OK(root.ToGraphDef(graph_def));
57 GraphDef graph_def; local
58 CreateTestGraph(root, &input, &output_name, &graph_def);
60 graph_def.SerializeToString(&graph_def_serialized);
85 GraphDef graph_def; local
86 CreateTestGraph(root, &input, &output_name, &graph_def);
87 TF_ASSERT_OK(WriteTextProto(Env::Default(), filename_txt, graph_def));
|
