Searched refs:num_retvals_per_replica_ (Results 1 – 1 of 1) sorted by relevance
157 num_retvals_per_replica_(num_retvals_per_replica) {} in ParameterInfo()171 int64 NumRetvalsPerReplica() const { return num_retvals_per_replica_; } in NumRetvalsPerReplica()217 return num_replicas_ * num_retvals_per_replica_; in NumOutputsToHost()243 int64 num_retvals_per_replica_ = 0; variable