Searched refs:data_outputs (Results 1 – 2 of 2) sorted by relevance
292 std::vector<std::vector<std::pair<Node*, int>>> data_outputs(num_outputs); in BuildXlaLaunchOps() local308 data_outputs[le->src_output()].push_back( in BuildXlaLaunchOps()355 for (int i = 0, end = data_outputs.size(); i < end; ++i) { in BuildXlaLaunchOps()356 for (const auto& successor : data_outputs[i]) { in BuildXlaLaunchOps()
2791 std::vector<std::vector<std::pair<Node*, int>>> data_outputs(num_replicas * in BuildTPUReplicateOps() local2805 data_outputs[pos].push_back({edge->dst(), edge->dst_input()}); in BuildTPUReplicateOps()2852 for (int i = 0; i < data_outputs.size(); ++i) { in BuildTPUReplicateOps()2853 for (const auto& successor : data_outputs[i]) { in BuildTPUReplicateOps()