Searched refs:num_replicas_ (Results 1 – 4 of 4) sorted by relevance
29 AutoParallel(int num_replicas) : num_replicas_(num_replicas) { in AutoParallel()30 CHECK(num_replicas_ >= 2); in AutoParallel()49 int num_replicas_; variable
45 tensor->add_float_val(static_cast<float>(num_replicas_)); in AddNodeDivConst()240 for (int i = 0; i < num_replicas_; i++) { in BuildGraph()245 for (int j = 0; j < num_replicas_; j++) { in BuildGraph()
72 int num_replicas() const { return num_replicas_; } in num_replicas()81 int num_replicas_ = 1; variable
63 num_replicas_ = num_replicas; in set_num_replicas()75 device_ordinal_, result_layout, num_replicas_); in ToString()