Home
last modified time | relevance | path

Searched refs:graph_copy (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_xla_computations_pass_test.cc253 std::unique_ptr<Graph> graph_copy(new Graph(&flib_def)); in TEST() local
254 CopyGraph(*graph, graph_copy.get()); in TEST()
292 EncapsulateXlaComputationsPass::Encapsulate(&graph_copy, &flib_def)); in TEST()
294 graph_copy->BuildNodeNameIndex(); in TEST()
/external/tensorflow/tensorflow/core/grappler/costs/
Danalytical_cost_estimator.cc161 GraphDef graph_copy = optimized_graph; in PredictCosts() local
163 item_->WithGraph(std::move(graph_copy))); in PredictCosts()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler_test.cc597 std::unique_ptr<Graph> graph_copy(new Graph(OpRegistry::Global())); in TEST_F() local
598 CopyGraph(*graph, graph_copy.get()); in TEST_F()
603 std::move(graph_copy), args, &result)); in TEST_F()
1486 std::unique_ptr<Graph> graph_copy(new Graph(OpRegistry::Global())); in TEST_F() local
1487 CopyGraph(*graph, graph_copy.get()); in TEST_F()
1490 std::move(graph_copy), args, &result)); in TEST_F()
1537 std::unique_ptr<Graph> graph_copy(new Graph(OpRegistry::Global())); in TEST_F() local
1538 CopyGraph(*graph, graph_copy.get()); in TEST_F()
1540 TF_ASSERT_OK(compiler.CompileGraph(options, "NoOp", std::move(graph_copy), in TEST_F()
1555 std::unique_ptr<Graph> graph_copy(new Graph(OpRegistry::Global())); in TEST_F() local
[all …]
/external/tensorflow/tensorflow/python/training/
Dmoving_averages_test.py474 graph_copy = ops.Graph()
475 with graph_copy.as_default():
477 return graph_copy
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_session_test.cc248 GraphDef graph_copy(graph); in TEST() local
249 graph::SetDefaultDevice(cluster->devices()[0].name(), &graph_copy); in TEST()
250 TF_CHECK_OK(session->Create(graph_copy)); in TEST()
255 GraphDef graph_copy(graph); in TEST() local
256 graph::SetDefaultDevice(cluster->devices()[1].name(), &graph_copy); in TEST()
257 auto status = session->Create(graph_copy); in TEST()