Home
last modified time | relevance | path

Searched refs:copied_node (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_op_test.cc136 NodeDef& copied_node = *execute_info.mutable_remote_graph()->add_node(); in BuildRemoteFusedGraphExecuteInfo() local
137 copied_node = node; in BuildRemoteFusedGraphExecuteInfo()
142 std::vector<TensorShape>({TensorShape()}), &copied_node); in BuildRemoteFusedGraphExecuteInfo()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmemory_optimizer.cc396 NodeDef* copied_node = graph->add_node(); in RecomputeSubgraph() local
397 copied_node->set_name( in RecomputeSubgraph()
399 copied_node->set_op(original_node->op()); in RecomputeSubgraph()
400 *copied_node->mutable_attr() = original_node->attr(); in RecomputeSubgraph()
401 copied_node->set_device(original_node->device()); in RecomputeSubgraph()
405 *copied_node->add_input() = RecomputedOrOriginalNodeName( in RecomputeSubgraph()
410 *copied_node->add_input() = in RecomputeSubgraph()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dtransform_utils_test.cc168 NodeDef copied_node; in TestCopyNodeAttr() local
169 CopyNodeAttr(node, "foo", "bar", &copied_node); in TestCopyNodeAttr()
170 EXPECT_EQ(3, copied_node.attr().at("bar").i()); in TestCopyNodeAttr()