Home
last modified time | relevance | path

Searched refs:old_num_replicated_outputs (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc1057 int old_num_replicated_outputs = output_types.size() / num_replicas; in MoveTailOutsideCompilationToHost() local
1059 old_num_replicated_outputs + oc_input_edges.size(); in MoveTailOutsideCompilationToHost()
1060 VLOG(5) << "old_num_replicated_outputs: " << old_num_replicated_outputs; in MoveTailOutsideCompilationToHost()
1066 for (int i = 0; i < old_num_replicated_outputs; i++) { in MoveTailOutsideCompilationToHost()
1069 for (int i = old_num_replicated_outputs; i < new_num_replicated_outputs; in MoveTailOutsideCompilationToHost()
1071 new_output_types.push_back(new_ret_types[i - old_num_replicated_outputs]); in MoveTailOutsideCompilationToHost()
1084 old_num_replicated_outputs * num_replicas); in MoveTailOutsideCompilationToHost()
1088 e->src_output() < old_num_replicated_outputs * num_replicas) { in MoveTailOutsideCompilationToHost()
1098 for (int output_index = 0; output_index < old_num_replicated_outputs; in MoveTailOutsideCompilationToHost()
1101 replicated_outputs[replica_id * old_num_replicated_outputs + in MoveTailOutsideCompilationToHost()
[all …]