Searched refs:num_distributed_args_ (Results 1 – 1 of 1) sorted by relevance
155 num_distributed_args_(num_distributed_args), in ParameterInfo()165 int64_t NumDistributedArgs() const { return num_distributed_args_; } in NumDistributedArgs()181 index < (num_per_replica_args_ + num_distributed_args_); in IsDistributedArg()185 return (index >= num_per_replica_args_ + num_distributed_args_) && in IsBroadcastArg()186 index < (num_per_replica_args_ + num_distributed_args_ + in IsBroadcastArg()191 return index >= (num_per_replica_args_ + num_distributed_args_ + in IsVariableArg()193 index < (num_per_replica_args_ + num_distributed_args_ + in IsVariableArg()198 return index >= (num_per_replica_args_ + num_distributed_args_ + in IsConstantArg()200 index < (num_per_replica_args_ + num_distributed_args_ + in IsConstantArg()207 return num_replicas_ * num_per_replica_args_ + num_distributed_args_ + in NumInputsFromHost()[all …]