Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dencapsulate_tpu_computations_pass.cc1027 int new_num_replicated_outputs = in MoveTailOutsideCompilationToHost() local
1030 VLOG(5) << "new_num_replicated_outputs: " << new_num_replicated_outputs; in MoveTailOutsideCompilationToHost()
1038 for (int i = old_num_replicated_outputs; i < new_num_replicated_outputs; in MoveTailOutsideCompilationToHost()
1075 replica_id * new_num_replicated_outputs + output_index, dst, in MoveTailOutsideCompilationToHost()
1142 int src_output = replica_id * new_num_replicated_outputs + in MoveTailOutsideCompilationToHost()
1150 for (int i = old_num_replicated_outputs; i < new_num_replicated_outputs; in MoveTailOutsideCompilationToHost()