Home
last modified time | relevance | path

Searched refs:replica_count_ (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_config.h81 replica_count_ = replica_count; in set_replica_count()
83 int64 replica_count() const { return replica_count_; } in replica_count()
128 int64 replica_count_ = 1; variable
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_all_reduce_thunk.h53 const int64 replica_count_;
Dnccl_all_reduce_thunk.cc320 participant.replica_count = replica_count_; in ExecuteOnStream()
350 replica_count_(replica_count), in NcclAllReduceThunk()