Searched refs:num_distributed_args_ (Results 1 – 1 of 1) sorted by relevance
153 num_distributed_args_(num_distributed_args), in ParameterInfo()163 int64 NumDistributedArgs() const { return num_distributed_args_; } in NumDistributedArgs()179 index < (num_per_replica_args_ + num_distributed_args_); in IsDistributedArg()184 index < (num_per_replica_args_ + num_distributed_args_ + in IsBroadcastArg()190 index < (num_per_replica_args_ + num_distributed_args_ + in IsVariableArg()195 return index >= (num_per_replica_args_ + num_distributed_args_ + in IsConstantArg()197 index < (num_per_replica_args_ + num_distributed_args_ + in IsConstantArg()204 return num_replicas_ * num_per_replica_args_ + num_distributed_args_ + in NumInputsFromHost()210 return num_per_replica_args_ + num_distributed_args_ + in NumInputsToEachReplica()224 return num_replicas_ * num_per_replica_args_ + num_distributed_args_; in FirstBroadcastArgFromHost()[all …]