Home
last modified time | relevance | path

Searched defs:num_centers (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dclustering_ops.cc304 const int64_t num_centers = centers_tensor.dim_size(0); in Compute() local
472 const int64_t num_centers = centers.rows(); in FindKNearestCentersOneBlock() local
503 const int64_t num_centers = centers.rows(); in FindKNearestCentersBlockwise() local
Dclustering_ops_test.cc172 Graph* SetUpNearestNeighbors(int num_dims, int num_points, int num_centers, in SetUpNearestNeighbors()