Home
last modified time | relevance | path

Searched refs:num_replicas_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_parallel.h29 AutoParallel(int num_replicas) : num_replicas_(num_replicas) { in AutoParallel()
30 CHECK(num_replicas_ >= 2); in AutoParallel()
49 int num_replicas_; variable
Dauto_parallel.cc45 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()
/external/tensorflow/tensorflow/compiler/xla/client/
Dexecutable_build_options.h72 int num_replicas() const { return num_replicas_; } in num_replicas()
81 int num_replicas_ = 1; variable
Dexecutable_build_options.cc63 num_replicas_ = num_replicas; in set_num_replicas()
75 device_ordinal_, result_layout, num_replicas_); in ToString()