Searched defs:num_retries_per_sample (Results 1 – 2 of 2) sorted by relevance
57 Tensor num_retries_per_sample(DT_INT64, TensorShape({})); in SetUpKmeansPlusPlusInitialization() local
105 const int64_t num_retries_per_sample = [&]() { in Compute() local