Searched refs:replica_count_ (Results 1 – 4 of 4) sorted by relevance
32 : replica_count_(replica_count) {} in AllReduceSimplifier()41 int64 replica_count_;
127 replica_count_ = replica_count; in set_replica_count()129 int64 replica_count() const { return replica_count_; } in replica_count()254 int64 replica_count_ = 1; variable
41 return replica_count_; in Run()
42 : num_cores_(0), replica_count_(replica_count) {} in TpuAotCompilationOptions()51 int64 replica_count() const override { return replica_count_; } in replica_count()72 int64 replica_count_; variable