Home
last modified time | relevance | path

Searched refs:new_graph (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/core/tfrt/utils/
Dgraph_partition.cc433 auto new_graph = std::make_unique<Graph>(graph->flib_def()); in InsertTransferOps() local
454 &new_graph->flib_def(), absl::StrCat(graph_func_name, "-partition-", in InsertTransferOps()
471 auxiliary_output_node, subgraph, new_graph.get())); in InsertTransferOps()
479 control_outputs, subgraph, new_graph.get())); in InsertTransferOps()
517 &new_graph->flib_def(), in InsertTransferOps()
524 stateful_call_func_name, host_device, new_graph.get())); in InsertTransferOps()
537 .Finalize(new_graph.get(), nullptr)); in InsertTransferOps()
548 .Finalize(new_graph.get(), nullptr)); in InsertTransferOps()
551 return new_graph; in InsertTransferOps()
Dgraph_partition_test.cc90 std::unique_ptr<Graph> new_graph, in TEST_F()
97 new_graph->ToGraphDef(&new_graphdef); in TEST_F()
139 std::unique_ptr<Graph> new_graph, in TEST_F()
146 new_graph->ToGraphDef(&new_graphdef); in TEST_F()
Dtfrt_graph_execution_state.cc390 std::unique_ptr<Graph> new_graph, in BuildXlaOpsAndMaybeInsertTransferOps()
395 DumpGraphToFile("after_transfer_ops_insertion", *new_graph); in BuildXlaOpsAndMaybeInsertTransferOps()
398 return new_graph; in BuildXlaOpsAndMaybeInsertTransferOps()
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_execution_state.cc609 Status GraphExecutionState::InitBaseGraph(std::unique_ptr<Graph>&& new_graph) { in InitBaseGraph() argument
611 RestoreStatefulNodes(new_graph.get()); in InitBaseGraph()
616 optimization_options.graph = &new_graph; in InitBaseGraph()
623 Placer placer(new_graph.get(), "", flib_def_.get(), device_set_, in InitBaseGraph()
635 for (const Node* n : new_graph->nodes()) { in InitBaseGraph()
640 SaveStatefulNodes(new_graph.get()); in InitBaseGraph()
641 graph_ = new_graph.release(); in InitBaseGraph()
809 GraphDef new_graph; in OptimizeGraph()
812 cpu_device, &cluster, &new_graph)); in OptimizeGraph()
820 for (const FunctionDef& fdef : new_graph.library().function()) { in OptimizeGraph()
[all …]
Dpartitioning_utils_test.cc191 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<Graph> new_graph, in TEST_F()
195 EXPECT_EQ(new_graph->num_op_nodes(), 3); in TEST_F()
197 for (const auto* op : new_graph->op_nodes()) { in TEST_F()
236 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<Graph> new_graph, in TEST_F()
240 EXPECT_EQ(new_graph->num_op_nodes(), 7); in TEST_F()
242 for (const auto* op : new_graph->op_nodes()) { in TEST_F()
Dpartitioning_utils.cc126 auto new_graph = std::make_unique<Graph>(graph->flib_def()); in InsertTransferOps() local
150 new_graph.get())); in InsertTransferOps()
151 return std::move(new_graph); in InsertTransferOps()
Dshape_refiner.cc173 Graph* new_graph = new Graph(function_library_); in InferShapesForFunction() local
177 ConvertNodeDefsToGraph(options, result.nodes, new_graph)); in InferShapesForFunction()
178 functions_[function_def].reset(new_graph); in InferShapesForFunction()
179 graph = new_graph; in InferShapesForFunction()
/external/tensorflow/tensorflow/lite/python/
Dlite_flex_test.py213 new_graph = graph_pb2.GraphDef()
222 new_graph.CopyFrom(sess.graph_def)
225 for node in new_graph.node:
233 return (new_graph, inputs, outputs)
236 new_graph, inputs, outputs = self._createGraphWithCustomOp(
243 import_graph_def(new_graph, name='')
Dmetrics_nonportable_test.py450 new_graph = graph_pb2.GraphDef()
459 new_graph.CopyFrom(sess.graph_def)
462 for node in new_graph.node:
470 return (new_graph, inputs, outputs)
472 new_graph, inputs, outputs = create_graph_with_custom_add()
478 import_graph_def(new_graph, name='')
Dtflite_convert_test.py247 new_graph = graph_pb2.GraphDef()
256 new_graph.CopyFrom(sess.graph_def)
259 for node in new_graph.node:
271 import_graph_def(new_graph, name='')
/external/tensorflow/tensorflow/lite/python/metrics/
Dmetrics_nonportable_test.py452 new_graph = graph_pb2.GraphDef()
461 new_graph.CopyFrom(sess.graph_def)
464 for node in new_graph.node:
472 return (new_graph, inputs, outputs)
474 new_graph, inputs, outputs = create_graph_with_custom_add()
480 import_graph_def(new_graph, name='')
/external/tensorflow/tensorflow/python/eager/
Dwrap_function_test.py514 new_graph = lambda: wrap_function.WrappedGraph(variable_holder=vh) function
516 add = new_graph().wrap_function(add_v1, signature)
517 subtract = new_graph().wrap_function(subtract_v1, signature)
518 different_variable_fn = new_graph().wrap_function(
520 increment_variable = new_graph().wrap_function(
/external/tensorflow/tensorflow/dtensor/cc/
Ddtensor_device.cc1170 auto new_graph = std::make_unique<Graph>(graph.flib_def()); in SelectGraphToExecute() local
1171 CopyGraph(graph, new_graph.get()); in SelectGraphToExecute()
1174 for (Node* node : new_graph->nodes()) { in SelectGraphToExecute()
1180 for (Node* node : new_graph->nodes()) { in SelectGraphToExecute()
1201 new_graph->RemoveNode(n); in SelectGraphToExecute()
1220 new_graph->RemoveNode(arg_node); in SelectGraphToExecute()
1225 for (Node* n : new_graph->nodes()) { in SelectGraphToExecute()
1250 *new_graph); in SelectGraphToExecute()
1252 return new_graph; in SelectGraphToExecute()
1272 std::unique_ptr<Graph> new_graph, in AddExecutionFunctionDefsToFunctionDefLibrary()
[all …]
/external/tensorflow/tensorflow/compiler/jit/
Dmark_for_compilation_pass.cc1577 Graph new_graph(graph_->op_registry()); in DumpPostClusteringGraphs() local
1578 CopyGraph(*graph_, &new_graph); in DumpPostClusteringGraphs()
1580 for (Node* n : new_graph.nodes()) { in DumpPostClusteringGraphs()
1594 DumpGraphToFile("mark_for_compilation_annotated", new_graph, flib_def_); in DumpPostClusteringGraphs()
/external/tensorflow/tensorflow/core/graph/
Dgraph.cc440 std::unique_ptr<Graph> new_graph(new Graph(flib_def())); in Clone() local
441 new_graph->Copy(*this); in Clone()
442 return new_graph; in Clone()
/external/tensorflow/tensorflow/python/framework/
Dmeta_graph_test.py98 new_graph = ops.Graph()
99 with self.session(graph=new_graph) as sess:
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer_test.cc3544 GraphDef new_graph; in TEST_F() local
3546 OptimizeAndPrune(&optimizer, &item, &new_graph); in TEST_F()
3549 EXPECT_EQ(CountOpNodes(new_graph, "Relu"), 2); in TEST_F()
3551 auto tensors = EvaluateNodes(new_graph, item.fetch); in TEST_F()
4213 GraphDef new_graph; in TEST_F() local
4216 OptimizeTwice(&optimizer, &item, &new_graph); in TEST_F()
4219 VerifyGraphsMatch(item.graph, new_graph, __LINE__); in TEST_F()
4221 auto tensors = EvaluateNodes(new_graph, item.fetch); in TEST_F()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc630 Status UpdateFunctionLibDefinition(const Graph& new_graph, in UpdateFunctionLibDefinition() argument
634 TF_RETURN_IF_ERROR(GraphToFunctionDef(new_graph, function_name, &graph_fdef)); in UpdateFunctionLibDefinition()