Searched refs:src_replicated_nodes (Results 1 – 1 of 1) sorted by relevance
82 const std::vector<Node*>& src_replicated_nodes = in ReplicateFromCompositeDeviceToCompositeDevice() local86 if (src_replicated_nodes.size() != dst_replicated_nodes.size()) { in ReplicateFromCompositeDeviceToCompositeDevice()90 edge->src()->name(), " (", src_replicated_nodes.size(), in ReplicateFromCompositeDeviceToCompositeDevice()94 for (int i = 0; i < src_replicated_nodes.size(); ++i) { in ReplicateFromCompositeDeviceToCompositeDevice()101 graph->AddEdge(src_replicated_nodes.at(i), edge->src_output(), dst, in ReplicateFromCompositeDeviceToCompositeDevice()114 const std::vector<Node*>& src_replicated_nodes = in ReplicateFromCompositeDeviceToRegularDevice() local123 graph->AddEdge(src_replicated_nodes.at(i), edge->src_output(), dst, in ReplicateFromCompositeDeviceToRegularDevice()135 for (Node* replicated_node : src_replicated_nodes) { in ReplicateFromCompositeDeviceToRegularDevice()156 const int num_replicas = src_replicated_nodes.size(); in ReplicateFromCompositeDeviceToRegularDevice()161 inputs.reserve(src_replicated_nodes.size()); in ReplicateFromCompositeDeviceToRegularDevice()[all …]