Searched refs:old_num_per_replica_inputs (Results 1 – 1 of 1) sorted by relevance
460 int old_num_per_replica_inputs = in MoveHeadOutsideCompilationToHost() local462 VLOG(5) << "old_num_per_replica_inputs: " << old_num_per_replica_inputs; in MoveHeadOutsideCompilationToHost()486 (index < old_num_per_replica_inputs) in MoveHeadOutsideCompilationToHost()487 ? (old_num_per_replica_inputs * replica_id + index) in MoveHeadOutsideCompilationToHost()488 : (old_num_per_replica_inputs * num_replicas + in MoveHeadOutsideCompilationToHost()489 (index - old_num_per_replica_inputs)); in MoveHeadOutsideCompilationToHost()531 old_num_per_replica_inputs + oc_output_edges.size(); in MoveHeadOutsideCompilationToHost()555 for (int i = 0; i < old_num_per_replica_inputs; ++i) { in MoveHeadOutsideCompilationToHost()558 for (int i = old_num_per_replica_inputs; i < new_num_per_replica_inputs; in MoveHeadOutsideCompilationToHost()560 new_input_types.push_back(new_arg_types[i - old_num_per_replica_inputs]); in MoveHeadOutsideCompilationToHost()[all …]