Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_op_test.cc129 const GraphDef& original_graph) { in BuildRemoteFusedGraphExecuteInfo() argument
135 for (const NodeDef& node : original_graph.node()) { in BuildRemoteFusedGraphExecuteInfo()
251 static Status RewriteGraphToFusedGraph(const GraphDef& original_graph, in RewriteGraphToFusedGraph() argument
258 BuildRemoteFusedGraphExecuteInfo(original_graph); in RewriteGraphToFusedGraph()
282 GraphDef original_graph; in TEST() local
284 NAME_A, NODE_A_VAL, NAME_B, NODE_B_VAL, NAME_A_PLUS_B, &original_graph)); in TEST()
288 TF_ASSERT_OK(RewriteGraphToFusedGraph(original_graph, &fused_graph)); in TEST()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Ddependency_optimizer_test.cc36 void VerifyGraphsEqual(const GraphDef& original_graph, in VerifyGraphsEqual() argument
38 EXPECT_EQ(original_graph.node_size(), optimized_graph.node_size()) << func; in VerifyGraphsEqual()
39 for (int i = 0; i < original_graph.node_size(); ++i) { in VerifyGraphsEqual()
40 const NodeDef& original = original_graph.node(i); in VerifyGraphsEqual()
Dloop_optimizer_test.cc501 void VerifyGraphsEqual(const GraphDef& original_graph, in VerifyGraphsEqual() argument
503 EXPECT_EQ(original_graph.node_size(), optimized_graph.node_size()) << func; in VerifyGraphsEqual()
504 for (int i = 0; i < original_graph.node_size(); ++i) { in VerifyGraphsEqual()
505 const NodeDef& original = original_graph.node(i); in VerifyGraphsEqual()
Darithmetic_optimizer_test.cc74 void VerifyGraphsMatch(const GraphDef& original_graph, in VerifyGraphsMatch() argument
76 EXPECT_EQ(original_graph.node_size(), optimized_graph.node_size()) << line; in VerifyGraphsMatch()
77 for (int i = 0; i < original_graph.node_size(); ++i) { in VerifyGraphsMatch()
78 const NodeDef& original = original_graph.node(i); in VerifyGraphsMatch()
/external/tensorflow/tensorflow/python/framework/
Dmeta_graph_test.py967 with ops.Graph().as_default() as original_graph:
971 graph=original_graph,