Embedding Projector: perform consistent knn calculation

tensorflow · Apr 7, 2023 · 5bd37ea · 5bd37ea
1 parent e5d1771
commit 5bd37ea
Showing 1 changed file with 6 additions and 10 deletions.
diff --git a/tensorboard/plugins/projector/vz_projector/data.ts b/tensorboard/plugins/projector/vz_projector/data.ts
@@ -86,7 +86,7 @@ export interface DataPoint {
     [key: string]: number;
   };
 }
-const IS_FIREFOX = navigator.userAgent.toLowerCase().indexOf('firefox') >= 0;
+
 /** Controls whether nearest neighbors computation is done on the GPU or CPU. */
 export const TSNE_SAMPLE_SIZE = 10000;
 export const UMAP_SAMPLE_SIZE = 5000;
@@ -473,15 +473,11 @@ export class DataSet {
           .map((neighbors) => neighbors.slice(0, nNeighbors))
       );
     } else {
-      const knnGpuEnabled = (await util.hasWebGLSupport()) && !IS_FIREFOX;
-      const result = await (knnGpuEnabled
-        ? knn.findKNNGPUCosDistNorm(data, nNeighbors, (d) => d.vector)
-        : knn.findKNN(
-            data,
-            nNeighbors,
-            (d) => d.vector,
-            (a, b) => vector.cosDistNorm(a, b)
-          ));
+      const result = await knn.findKNNGPUCosDistNorm(
+        data,
+        nNeighbors,
+        (d) => d.vector
+      );
       this.nearest = result;
       return Promise.resolve(result);
     }