Searched refs:new_num_per_replica_inputs (Results 1 – 1 of 1) sorted by relevance
530 int new_num_per_replica_inputs = in MoveHeadOutsideCompilationToHost() local532 VLOG(5) << "new_num_per_replica_inputs: " << new_num_per_replica_inputs; in MoveHeadOutsideCompilationToHost()552 new_input_types.reserve(num_replicas * new_num_per_replica_inputs + in MoveHeadOutsideCompilationToHost()558 for (int i = old_num_per_replica_inputs; i < new_num_per_replica_inputs; in MoveHeadOutsideCompilationToHost()591 num_replicas * new_num_per_replica_inputs + num_other_inputs); in MoveHeadOutsideCompilationToHost()593 num_replicas * new_num_per_replica_inputs + num_other_inputs - 1; in MoveHeadOutsideCompilationToHost()630 replica_id * new_num_per_replica_inputs + input_index); in MoveHeadOutsideCompilationToHost()634 input_index < new_num_per_replica_inputs; input_index++) { in MoveHeadOutsideCompilationToHost()643 replica_id * new_num_per_replica_inputs + input_index); in MoveHeadOutsideCompilationToHost()662 for (int i = old_num_per_replica_inputs; i < new_num_per_replica_inputs; in MoveHeadOutsideCompilationToHost()[all …]