Searched refs:points_tensor (Results 1 – 1 of 1) sorted by relevance
84 const Tensor& points_tensor = context->input(0); in Compute() local89 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(points_tensor.shape()), in Compute()101 const int64_t num_points = points_tensor.dim_size(0); in Compute()102 const int64_t point_dimensions = points_tensor.dim_size(1); in Compute()126 points_tensor.matrix<float>().data(), num_points, point_dimensions); in Compute()291 const Tensor& points_tensor = context->input(0); in Compute() local295 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(points_tensor.shape()), in Compute()302 const int64_t num_points = points_tensor.dim_size(0); in Compute()303 const int64_t point_dimensions = points_tensor.dim_size(1); in Compute()315 points_tensor.matrix<float>().data(), num_points, point_dimensions); in Compute()