[edgeconf] knn: adapted naming

38353d79 · Dennis Noll · 2d34266f · 38353d79
Commit 38353d79 authored Mar 23, 2020 by Dennis Noll
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

edgeconv.py edgeconv.py +2 -2

No files found.
--- a/edgeconv.py
+++ b/edgeconv.py
@@ -147,5 +147,5 @@ def knn(topk_indices, features):
    # features: (N, P, C)
    # return: (N, P, K, C)
    with tf.name_scope("knn"):
-        knn_features = tf.gather(features, topk_indices, batch_dims=1)
-    return knn_features
+        topk_features = tf.gather(features, topk_indices, batch_dims=1)
+        return topk_features