Searched refs:loss_repeated_mean (Results 1 – 7 of 7) sorted by relevance
74 void set_loss_repeated_mean(bool loss_repeated_mean);75 bool loss_repeated_mean() const { return loss_repeated_mean_; } in loss_repeated_mean() function
104 …ParallelContext::set_loss_repeated_mean(bool loss_repeated_mean) { loss_repeated_mean_ = loss_repe… in set_loss_repeated_mean() argument
158 def set_loss_repeated_mean(self, loss_repeated_mean): argument170 if not isinstance(loss_repeated_mean, bool):173 self._context_handle.set_loss_repeated_mean(loss_repeated_mean)681 loss_repeated_mean=bool, parallel_mode=str, auto_parallel_search_mode=str,
101 if (!ParallelContext::GetInstance()->loss_repeated_mean()) { in InferAsLossDivisor()
177 if (!ParallelContext::GetInstance()->loss_repeated_mean()) { in InferAsLossDivisor()
1592 if (!ParallelContext::GetInstance()->loss_repeated_mean()) { in InferAsLossDivisor()
147 .def("get_loss_repeated_mean", &ParallelContext::loss_repeated_mean, "Get loss repeated mean.") in PYBIND11_MODULE()