Searched refs:is_per_replica_arg (Results 1 – 1 of 1) sorted by relevance
1996 const bool is_per_replica_arg = params_info.IsPerReplicaArg(i); in AssignArgsAndRetvalsToCores() local1997 if (is_per_replica_arg || params_info.IsDistributedArg(i)) { in AssignArgsAndRetvalsToCores()2000 i + (is_per_replica_arg ? 0 : index_offset), &input_node)); in AssignArgsAndRetvalsToCores()