Searched refs:num_to_sample (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_KmeansPlusPlusInitialization.pbtxt | 11 name: "num_to_sample"
28 heuristic is used to sample O(log(num_to_sample)) additional points.
34 Matrix of shape (num_to_sample, d). The sampled rows.
37 summary: "Selects num_to_sample rows of input using the KMeans++ criterion."
41 distance from the nearest row selected thus far till num_to_sample rows have
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | clustering_ops_test.cc | 51 int num_to_sample, in SetUpKmeansPlusPlusInitialization() argument
59 sample_size.flat<int64>().setConstant(num_to_sample); in SetUpKmeansPlusPlusInitialization()
73 template <int num_points, int num_to_sample, int num_dims,
77 num_dims, num_points, num_to_sample, retries_per_sample); in BM_KmeansPlusPlusInitialization()
80 num_dims * num_to_sample); in BM_KmeansPlusPlusInitialization()
133 template <int num_points, int num_to_sample, int num_dims>
138 num_dims * num_to_sample); in BM_KMC2Initialization()
|
| D | clustering_ops.cc | 103 const int64_t num_to_sample = num_to_sample_tensor.scalar<int64>()(); in Compute() local
108 : 2 + static_cast<int64>(std::log(num_to_sample)); in Compute()
113 OP_REQUIRES(context, num_to_sample > 0, in Compute()
115 OP_REQUIRES(context, num_to_sample <= num_points, in Compute()
117 num_to_sample, " vs ", num_points, ".")); in Compute()
122 0, TensorShape({num_to_sample, point_dimensions}), in Compute()
131 output_sampled_points_tensor->matrix<float>().data(), num_to_sample, in Compute()
195 for (int64_t i = 0; i < num_to_sample; ++i) { in Compute()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | KmeansPlusPlusInitialization.pbtxt | 8 name: "num_to_sample"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | KmeansPlusPlusInitialization.pbtxt | 8 name: "num_to_sample"
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | clustering_ops.py | 650 num_to_sample = math_ops.maximum(
655 return math_ops.less(i, num_to_sample)
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 37672 func KmeansPlusPlusInitialization(scope *Scope, points tf.Output, num_to_sample tf.Output, seed tf.…
37679 points, num_to_sample, seed, num_retries_per_sample,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 20764 name: "num_to_sample"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 30779 name: "num_to_sample"
|