Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/python/
Dlite_flex_test.py155 new_graph = graph_pb2.GraphDef()
164 new_graph.CopyFrom(sess.graph_def)
167 for node in new_graph.node:
175 return (new_graph, inputs, outputs)
178 new_graph, inputs, outputs = self._createGraphWithCustomOp(
185 import_graph_def(new_graph, name='')
Dtflite_convert_test.py250 new_graph = graph_pb2.GraphDef()
259 new_graph.CopyFrom(sess.graph_def)
262 for node in new_graph.node:
274 import_graph_def(new_graph, name='')
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_execution_state.cc606 Status GraphExecutionState::InitBaseGraph(std::unique_ptr<Graph>&& new_graph) { in InitBaseGraph() argument
608 RestoreStatefulNodes(new_graph.get()); in InitBaseGraph()
613 optimization_options.graph = &new_graph; in InitBaseGraph()
620 Placer placer(new_graph.get(), "", flib_def_.get(), device_set_, in InitBaseGraph()
632 for (const Node* n : new_graph->nodes()) { in InitBaseGraph()
637 SaveStatefulNodes(new_graph.get()); in InitBaseGraph()
638 graph_ = new_graph.release(); in InitBaseGraph()
804 GraphDef new_graph; in OptimizeGraph()
807 cpu_device, &cluster, &new_graph)); in OptimizeGraph()
815 for (const FunctionDef& fdef : new_graph.library().function()) { in OptimizeGraph()
[all …]
Dshape_refiner.cc156 Graph* new_graph = new Graph(function_library_); in InferShapesForFunction() local
160 ConvertNodeDefsToGraph(options, result.nodes, new_graph)); in InferShapesForFunction()
161 functions_[function_def].reset(new_graph); in InferShapesForFunction()
162 graph = new_graph; in InferShapesForFunction()
/external/tensorflow/tensorflow/python/eager/
Dwrap_function_test.py518 new_graph = lambda: wrap_function.WrappedGraph(variable_holder=vh) function
520 add = new_graph().wrap_function(add_v1, signature)
521 subtract = new_graph().wrap_function(subtract_v1, signature)
522 different_variable_fn = new_graph().wrap_function(
524 increment_variable = new_graph().wrap_function(
/external/tensorflow/tensorflow/compiler/jit/
Dmark_for_compilation_pass.cc1450 Graph new_graph(graph_->op_registry()); in DumpPostClusteringGraphs() local
1451 CopyGraph(*graph_, &new_graph); in DumpPostClusteringGraphs()
1453 for (Node* n : new_graph.nodes()) { in DumpPostClusteringGraphs()
1467 DumpGraphToFile("mark_for_compilation_annotated", new_graph, flib_def_); in DumpPostClusteringGraphs()
/external/tensorflow/tensorflow/python/framework/
Dmeta_graph_test.py102 new_graph = ops.Graph()
103 with self.session(graph=new_graph) as sess:
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer_test.cc3085 GraphDef new_graph; in TEST_F() local
3087 OptimizeAndPrune(&optimizer, &item, &new_graph); in TEST_F()
3090 EXPECT_EQ(CountOpNodes(new_graph, "Relu"), 2); in TEST_F()
3092 auto tensors = EvaluateNodes(new_graph, item.fetch); in TEST_F()
3709 GraphDef new_graph; in TEST_F() local
3712 OptimizeTwice(&optimizer, &item, &new_graph); in TEST_F()
3715 VerifyGraphsMatch(item.graph, new_graph, __LINE__); in TEST_F()
3717 auto tensors = EvaluateNodes(new_graph, item.fetch); in TEST_F()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc627 Status UpdateFunctionLibDefinition(const Graph& new_graph, in UpdateFunctionLibDefinition() argument
631 TF_RETURN_IF_ERROR(GraphToFunctionDef(new_graph, function_name, &graph_fdef)); in UpdateFunctionLibDefinition()