Searched refs:distances_tensor (Results 1 – 1 of 1) sorted by relevance
238 const Tensor& distances_tensor = context->input(0); in Compute() local240 OP_REQUIRES(context, TensorShapeUtils::IsVector(distances_tensor.shape()), in Compute()244 const int64_t num_points = distances_tensor.dim_size(0); in Compute()252 auto distances = distances_tensor.flat<float>(); in Compute()