Home
last modified time | relevance | path

Searched refs:ToGraphDef (Results 1 – 25 of 96) sorted by relevance

1234

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer_test.cc169 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
192 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
214 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
227 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
240 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
253 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
266 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
281 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
308 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
345 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
[all …]
Darithmetic_optimizer_test.cc71 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
105 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
134 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
170 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
196 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
218 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
244 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
272 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
306 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
345 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
[all …]
Ddependency_optimizer_test.cc76 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
111 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
173 TF_CHECK_OK(scope.ToGraphDef(&item.graph)); in TEST_F()
211 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
230 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
273 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
302 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
360 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
419 TF_CHECK_OK(scope.ToGraphDef(&item.graph)); in TEST_F()
460 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
[all …]
Dmodel_pruner_test.cc66 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
102 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
140 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
184 TF_CHECK_OK(scope.ToGraphDef(&item.graph)); in TEST_F()
209 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
295 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
327 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
Dmemory_optimizer_test.cc46 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
85 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
109 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
149 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
231 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
291 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
329 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
361 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
Dconstant_folding_test.cc44 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
90 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
195 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
322 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
412 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
466 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
512 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
560 TF_CHECK_OK(s.ToGraphDef(&item.graph)); in TEST_F()
601 TF_CHECK_OK(scope.ToGraphDef(&item.graph)); in TEST_F()
644 TF_CHECK_OK(scope.ToGraphDef(&item.graph)); in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction_test.cc532 TF_ASSERT_OK(s.ToGraphDef(&expected)); in TEST_F()
535 g->ToGraphDef(&actual); in TEST_F()
554 TF_ASSERT_OK(s.ToGraphDef(&expected)); in TEST_F()
557 g->ToGraphDef(&actual); in TEST_F()
592 TF_ASSERT_OK(s.ToGraphDef(&e2)); in TEST_F()
595 g->ToGraphDef(&actual); in TEST_F()
603 g->ToGraphDef(&actual); in TEST_F()
627 TF_ASSERT_OK(s.ToGraphDef(&expected)); in TEST_F()
630 g->ToGraphDef(&actual); in TEST_F()
669 TF_ASSERT_OK(s.ToGraphDef(&expected)); in TEST_F()
[all …]
Ddirect_session_test.cc82 test::graph::ToGraphDef(&graph, &def_); in Initialize()
246 test::graph::ToGraphDef(&graph, &def); in TEST_F()
258 test::graph::ToGraphDef(&graph, &def); in TEST_F()
317 test::graph::ToGraphDef(&g, &def); in TEST()
352 test::graph::ToGraphDef(&g, &def); in TEST()
418 test::graph::ToGraphDef(&g, &def); in TEST()
461 test::graph::ToGraphDef(&g, &def); in TEST()
479 test::graph::ToGraphDef(&g, &def); in TEST()
497 test::graph::ToGraphDef(&g, &def); in TEST()
528 test::graph::ToGraphDef(&g, &def); in TEST()
[all …]
Ddirect_session_with_tracking_alloc_test.cc69 test::graph::ToGraphDef(&graph, &def); in TEST()
134 test::graph::ToGraphDef(&g, &def); in TEST()
177 test::graph::ToGraphDef(&graph, &def); in TestHWAccelerator()
262 test::graph::ToGraphDef(&graph, &def); in TEST()
321 test::graph::ToGraphDef(&graph, &def); in TEST()
/external/tensorflow/tensorflow/core/graph/
Dgraph_partition_test.cc204 const GraphDef& ToGraphDef() { in ToGraphDef() function in tensorflow::__anon518f74ac0111::GraphPartitionTest
205 TF_EXPECT_OK(in_.ToGraphDef(&in_graph_def_)); in ToGraphDef()
211 TF_EXPECT_OK(scope_a_.ToGraphDef(&graph_def)); in ExpectMatchA()
218 TF_EXPECT_OK(scope_b_.ToGraphDef(&graph_def)); in ExpectMatchB()
243 Partition(ToGraphDef(), &partitions_); in TEST_F()
256 Partition(ToGraphDef(), &partitions_); in TEST_F()
277 Partition(ToGraphDef(), &partitions_); in TEST_F()
301 Partition(ToGraphDef(), &partitions_); in TEST_F()
324 Partition(ToGraphDef(), &partitions_); in TEST_F()
349 Partition(ToGraphDef(), &partitions_); in TEST_F()
[all …]
Dgraph_def_builder.cc67 Status GraphDefBuilder::ToGraphDef(GraphDef* graph_def) const { in ToGraphDef() function in tensorflow::GraphDefBuilder
69 graph_.ToGraphDef(graph_def); in ToGraphDef()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_constants_test.cc70 TF_ASSERT_OK(root.ToGraphDef(&graph_def)); in TestSimpleAdd()
107 TF_ASSERT_OK(root.ToGraphDef(&graph_def)); in TestOpExclusionAdd()
131 TF_ASSERT_OK(root.ToGraphDef(&graph_def)); in TestShapePropagation()
160 TF_ASSERT_OK(root.ToGraphDef(&graph_def)); in TestPreserveOutputShapes()
235 TF_ASSERT_OK(o_root.ToGraphDef(&o_graph_def)); in TestReplaceSendRecvs()
245 TF_ASSERT_OK(n_root.ToGraphDef(&n_graph_def)); in TestReplaceSendRecvs()
270 TF_ASSERT_OK(o_root.ToGraphDef(&o_graph_def)); in TestReplaceSendRecvsPrefixNames()
279 TF_ASSERT_OK(n_root.ToGraphDef(&n_graph_def)); in TestReplaceSendRecvsPrefixNames()
317 TF_ASSERT_OK(root.ToGraphDef(&graph_def)); in TestRemoveUnusedNodes()
347 TF_ASSERT_OK(root.ToGraphDef(&graph_def)); in TestRemoveUnusedNodesMultipleOutputs()
Dquantize_nodes_test.cc179 TF_ASSERT_OK(root.ToGraphDef(&float_graph_def)); in TestIgnoreOps()
224 TF_ASSERT_OK(root.ToGraphDef(&float_graph_def)); in TestQuantizeMatMul()
275 TF_ASSERT_OK(root.ToGraphDef(&float_graph_def)); in TestQuantizeMul()
310 TF_ASSERT_OK(root.ToGraphDef(&float_graph_def)); in TestQuantizeAdd()
339 TF_ASSERT_OK(root.ToGraphDef(&float_graph_def)); in TestQuantizeConv2D()
362 TF_ASSERT_OK(root.ToGraphDef(&float_graph_def)); in TestQuantizeBiasAdd()
389 TF_ASSERT_OK(root.ToGraphDef(&float_graph_def)); in TestQuantizeConcat()
407 TF_ASSERT_OK(root.ToGraphDef(&float_graph_def)); in TestQuantizeRelu()
425 TF_ASSERT_OK(root.ToGraphDef(&float_graph_def)); in TestQuantizeRelu6()
444 TF_ASSERT_OK(root.ToGraphDef(&float_graph_def)); in TestQuantizeMaxPool()
[all …]
Dtransform_utils_test.cc53 TF_ASSERT_OK(root.ToGraphDef(&graph_def)); in TestMapNamesToNodes()
86 TF_ASSERT_OK(root.ToGraphDef(&graph_def)); in TestMapNodesToOutputs()
242 TF_ASSERT_OK(root.ToGraphDef(&graph_def)); in TestFilterGraphDef()
267 TF_ASSERT_OK(root.ToGraphDef(&graph_def)); in TestRemoveAttributes()
300 TF_ASSERT_OK(root.ToGraphDef(&graph_def)); in TestGetOpTypeMatches()
393 TF_ASSERT_OK(root.ToGraphDef(&graph_def)); in TestGetOpTypeMatchesDAG()
430 TF_ASSERT_OK(root.ToGraphDef(&graph_def)); in TestReplaceMatchingOpTypes()
539 TF_ASSERT_OK(root.ToGraphDef(&graph_def)); in TestRenameNodeInputs()
577 TF_ASSERT_OK(root.ToGraphDef(&graph_def)); in TestRenameNodeInputsWithRedirects()
615 TF_ASSERT_OK(root.ToGraphDef(&graph_def)); in TestRenameNodeInputsWithCycle()
[all …]
Dfold_batch_norms_test.cc67 TF_ASSERT_OK(root.ToGraphDef(&original_graph_def)); in TestFoldBatchNormsConv2D()
126 TF_ASSERT_OK(root.ToGraphDef(&original_graph_def)); in TestFoldBatchNormsConv2DShared()
175 TF_ASSERT_OK(root.ToGraphDef(&original_graph_def)); in TestFoldBatchNormsMatMul()
Dfuse_convolutions_test.cc76 TF_ASSERT_OK(root.ToGraphDef(&original_graph_def)); in TestFuseResizePadAndConv()
126 TF_ASSERT_OK(root.ToGraphDef(&original_graph_def)); in TestFuseResizeAndConv()
178 TF_ASSERT_OK(root.ToGraphDef(&original_graph_def)); in TestFusePadAndConv()
Dtransform_graph_test.cc76 TF_ASSERT_OK(root.ToGraphDef(&graph_def)); in TestConstantFolding()
132 TF_ASSERT_OK(root.ToGraphDef(&graph_def)); in TestTransformRegistration()
138 TF_ASSERT_OK(root.ToGraphDef(&graph_def)); in TestTransformRegistration()
Dinsert_logging_test.cc65 TF_ASSERT_OK(root.ToGraphDef(&graph_def)); in TestInsertLogging()
112 TF_ASSERT_OK(root.ToGraphDef(&graph_def)); in TestInsertLoggingByOpType()
161 TF_ASSERT_OK(root.ToGraphDef(&graph_def)); in TestInsertLoggingByPrefix()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow_test.cc99 graph.ToGraphDef(&graph_def); in TEST()
118 TF_EXPECT_OK(scope.ToGraphDef(&expected)); in TEST()
135 TF_EXPECT_OK(scope.ToGraphDef(&expected)); in TEST()
158 TF_EXPECT_OK(scope.ToGraphDef(&expected)); in TEST()
242 graph.ToGraphDef(&graph_def); in TEST()
256 TF_EXPECT_OK(scope.ToGraphDef(&expected)); in TEST()
270 TF_EXPECT_OK(scope.ToGraphDef(&expected)); in TEST()
291 TF_EXPECT_OK(scope.ToGraphDef(&expected)); in TEST()
345 graph.ToGraphDef(&graph_def); in TEST()
358 TF_EXPECT_OK(scope.ToGraphDef(&expected)); in TEST()
[all …]
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass_test.cc426 graph_out->ToGraphDef(&graphdef_out); in Encapsulate()
444 TF_EXPECT_OK(builder.ToGraphDef(&graphdef_in)); in TEST()
472 TF_EXPECT_OK(b1.ToGraphDef(&graphdef)); in TEST()
501 TF_EXPECT_OK(b2.ToGraphDef(&graphdef_expected)); in TEST()
527 TF_EXPECT_OK(b1.ToGraphDef(&graphdef)); in TEST()
566 TF_EXPECT_OK(b2.ToGraphDef(&graphdef_expected)); in TEST()
801 TF_EXPECT_OK(b1.ToGraphDef(&graphdef)); in TEST()
820 TF_EXPECT_OK(shape.ToGraphDef(&shape_graph)); in TEST()
872 TF_EXPECT_OK(b2.ToGraphDef(&graphdef_expected)); in TEST()
913 TF_EXPECT_OK(b1.ToGraphDef(&graphdef)); in TEST()
[all …]
Dgraph_to_functiondef_test.cc57 TF_EXPECT_OK(root.ToGraphDef(&graph_def)); in TEST()
93 TF_EXPECT_OK(root.ToGraphDef(&graph_def)); in TEST()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmaster_test.cc184 test::graph::ToGraphDef(&graph_1, &def_1); in TEST_F()
195 test::graph::ToGraphDef(&graph_2, &def_2); in TEST_F()
219 test::graph::ToGraphDef(&graph_1, &def_1); in TEST_F()
238 test::graph::ToGraphDef(&graph_1, &def_1); in TEST_F()
257 test::graph::ToGraphDef(&graph_1, &def_1); in TEST_F()
299 test::graph::ToGraphDef(&graph_0, &def_0); in TEST_F()
310 test::graph::ToGraphDef(&graph_1, &def_1); in TEST_F()
381 test::graph::ToGraphDef(&graph, &def); in TEST_F()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_session_test.cc60 test::graph::ToGraphDef(&graph, graph_def); in CreateGraphDef()
251 test::graph::ToGraphDef(&graph, &def); in FindMaxEigen()
319 test::graph::ToGraphDef(&graph, &def); in TEST()
399 test::graph::ToGraphDef(&graph, &def); in TEST()
434 test::graph::ToGraphDef(&graph, &def); in TEST()
491 test::graph::ToGraphDef(&graph, &def); in TEST()
555 test::graph::ToGraphDef(&g, &gdef); in TEST()
615 test::graph::ToGraphDef(&g, &gdef); in TEST()
656 test::graph::ToGraphDef(&g, &gdef); in TEST()
912 test::graph::ToGraphDef(&graph, &gdef); in TEST()
[all …]
/external/tensorflow/tensorflow/cc/tools/
Dfreeze_saved_model_test.cc186 TF_ASSERT_OK(scope.ToGraphDef(&graph_def)); in TEST_F()
211 TF_ASSERT_OK(scope.ToGraphDef(&graph_def)); in TEST_F()
228 TF_ASSERT_OK(expected_scope.ToGraphDef(&expected_graph_def)); in TEST_F()
246 TF_ASSERT_OK(scope.ToGraphDef(&graph_def)); in TEST_F()
283 TF_ASSERT_OK(scope.ToGraphDef(&graph_def)); in TEST_F()
/external/tensorflow/tensorflow/core/kernels/
Dimmutable_constant_op_test.cc99 TF_ASSERT_OK(root.ToGraphDef(&graph_def)); in TEST()
131 TF_ASSERT_OK(root.ToGraphDef(&graph_def)); in TEST()
172 TF_ASSERT_OK(root.ToGraphDef(&graph_def)); in TEST()

1234