Searched refs:kMaxIterations (Results 1 – 8 of 8) sorted by relevance
47 static constexpr int kMaxIterations = 1000; variable59 bool single_maxval, int64 kMaxIterations) { in TruncatedNormalKernel() argument60 const int32 max_samples_per_item = 2 * kMaxIterations; in TruncatedNormalKernel()137 while (numIterations < kMaxIterations) { in TruncatedNormalKernel()143 numIterations = kMaxIterations; in TruncatedNormalKernel()145 } else if (numIterations + 1 == kMaxIterations) { in TruncatedNormalKernel()149 numIterations = kMaxIterations; in TruncatedNormalKernel()167 while (numIterations < kMaxIterations) { in TruncatedNormalKernel()185 numIterations = kMaxIterations; in TruncatedNormalKernel()187 } else if (numIterations + 1 >= kMaxIterations) { in TruncatedNormalKernel()[all …]
48 static constexpr int kMaxIterations = 1000; variable81 gen_copy.Skip(start_batch * 2 * kMaxIterations * (samples_per_batch + 3) / in operator ()()166 if (numIterations > kMaxIterations) { in operator ()()175 << " kMaxIterations=" << kMaxIterations; in operator ()()204 if (accept || numIterations + 1 >= kMaxIterations) { in operator ()()251 if (accept || numIterations + 1 >= kMaxIterations) { in operator ()()425 generator_.ReserveSamples128(num_batches * 2 * functor::kMaxIterations * in Compute()
41 const uint32 kMaxIterations = 30; in EvaluateInverse() local50 for (uint32 iteration = 0; iteration < kMaxIterations; iteration++) in EvaluateInverse()
201 const uint32 kMaxIterations = 30; in EvaluateInverse() local212 for (uint32 iteration = 0; iteration < kMaxIterations; iteration++) in EvaluateInverse()
62 static const size_t kMaxIterations = 1000000000; variable260 const size_t next_iters = std::min(max_next_iters, kMaxIterations); in PredictNumItersNeeded()271 i.iters >= kMaxIterations || // Too many iterations already. in ShouldReportIterationResults()
905 final int kMaxIterations = 20; in ResolveReference() local908 in_out_value.get().data != 0 && iteration < kMaxIterations; in ResolveReference()
2202 constexpr int kMaxIterations = 50; in Optimize() local2212 if (iteration++ > kMaxIterations) { in Optimize()