Searched refs:replicated_node (Results 1 – 1 of 1) sorted by relevance
43 Node* replicated_node = graph->AddNode(node_def, &status); in ReplicateNode() local45 replicated_node->set_assigned_device_name(device); in ReplicateNode()46 if (replicated_node->IsArg()) { in ReplicateNode()47 replicated_node->AddAttr("sub_index", i); in ReplicateNode()49 replicated_nodes[i] = replicated_node; in ReplicateNode()100 for (Node* replicated_node : src_replicated_nodes) { in ReplicateFromCompositeDeviceToRegularDevice()101 graph->AddControlEdge(replicated_node, dst); in ReplicateFromCompositeDeviceToRegularDevice()106 for (Node* replicated_node : src_replicated_nodes) { in ReplicateFromCompositeDeviceToRegularDevice()107 if (replicated_node->assigned_device_name() == dst_device) { in ReplicateFromCompositeDeviceToRegularDevice()108 graph->AddEdge(replicated_node, edge->src_output(), dst, in ReplicateFromCompositeDeviceToRegularDevice()[all …]