Searched defs:num_centers (Results 1 – 2 of 2) sorted by relevance
304 const int64_t num_centers = centers_tensor.dim_size(0); in Compute() local472 const int64_t num_centers = centers.rows(); in FindKNearestCentersOneBlock() local503 const int64_t num_centers = centers.rows(); in FindKNearestCentersBlockwise() local
172 Graph* SetUpNearestNeighbors(int num_dims, int num_points, int num_centers, in SetUpNearestNeighbors()