Searched refs:num_per_replica_args_ (Results 1 – 1 of 1) sorted by relevance
152 num_per_replica_args_(num_per_replica_args), in ParameterInfo()161 int64 NumPerReplicaArgs() const { return num_per_replica_args_; } in NumPerReplicaArgs()174 return index < num_per_replica_args_; in IsPerReplicaArg()178 return index >= num_per_replica_args_ && in IsDistributedArg()179 index < (num_per_replica_args_ + num_distributed_args_); in IsDistributedArg()183 return index >= num_per_replica_args_ && in IsBroadcastArg()184 index < (num_per_replica_args_ + num_distributed_args_ + in IsBroadcastArg()189 return index >= (num_per_replica_args_ + num_broadcast_args_) && in IsVariableArg()190 index < (num_per_replica_args_ + num_distributed_args_ + in IsVariableArg()195 return index >= (num_per_replica_args_ + num_distributed_args_ + in IsConstantArg()[all …]