Searched refs:replicated_outputs (Results 1 – 3 of 3) sorted by relevance
1541 replicated_outputs = [[] for i in range(num_replicas)]1550 replicated_outputs[replica].append(None)1561 replicated_outputs[replica].append(1565 replicated_outputs = [1567 for replica_outs in replicated_outputs1570 return [compile_status, replicated_outputs]
1083 std::vector<std::vector<std::pair<Node*, int>>> replicated_outputs( in MoveTailOutsideCompilationToHost() local1089 replicated_outputs[e->src_output()].push_back( in MoveTailOutsideCompilationToHost()1101 replicated_outputs[replica_id * old_num_replicated_outputs + in MoveTailOutsideCompilationToHost()1137 replicated_outputs[replica_id * old_num_replicated_outputs + in MoveTailOutsideCompilationToHost()2773 std::vector<Node*> replicated_outputs(num_outputs); in BuildTPUReplicateOps() local2782 replicated_outputs[e->src_output()] = e->dst(); in BuildTPUReplicateOps()2796 TF_RET_CHECK(replicated_outputs[i] != nullptr); in BuildTPUReplicateOps()2797 for (const Edge* e : replicated_outputs[i]->out_edges()) { in BuildTPUReplicateOps()2807 output_types[pos] = replicated_outputs[i]->input_type(0); in BuildTPUReplicateOps()
277 Variadic<AnyType>:$replicated_outputs