Searched refs:src_replicated_nodes (Results 1 – 1 of 1) sorted by relevance
71 const std::vector<Node*>& src_replicated_nodes = in ReplicateFromCompositeDeviceToCompositeDevice() local75 if (src_replicated_nodes.size() != dst_replicated_nodes.size()) { in ReplicateFromCompositeDeviceToCompositeDevice()79 edge->src()->name(), " (", src_replicated_nodes.size(), in ReplicateFromCompositeDeviceToCompositeDevice()83 for (int i = 0; i < src_replicated_nodes.size(); ++i) { in ReplicateFromCompositeDeviceToCompositeDevice()84 graph->AddEdge(src_replicated_nodes.at(i), edge->src_output(), in ReplicateFromCompositeDeviceToCompositeDevice()96 const std::vector<Node*>& src_replicated_nodes = in ReplicateFromCompositeDeviceToRegularDevice() local100 for (Node* replicated_node : src_replicated_nodes) { in ReplicateFromCompositeDeviceToRegularDevice()106 for (Node* replicated_node : src_replicated_nodes) { in ReplicateFromCompositeDeviceToRegularDevice()127 const int num_replicas = src_replicated_nodes.size(); in ReplicateFromCompositeDeviceToRegularDevice()132 inputs.reserve(src_replicated_nodes.size()); in ReplicateFromCompositeDeviceToRegularDevice()[all …]