Fix benchmarks

DrA1ex · Sep 22, 2023 · d8af3ff · d8af3ff
1 parent b7d8ec0
commit d8af3ff
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/examples/src/benchmark.js b/examples/src/benchmark.js
@@ -39,8 +39,8 @@ const Count = 2000;
 const trainData = Matrix.random_2d(Count, Sizes[0]);
 const singleTrainData = trainData.slice(0, 1);
 
-const tfTrainData = tf.tensor(Matrix.copy_2d(trainData));
-const tfSingleData = tf.tensor(Matrix.copy_2d(singleTrainData));
+const tfTrainData = tf.tensor(trainData.map(t => Array.from(t)));
+const tfSingleData = tf.tensor(singleTrainData.map(t => Array.from(t)));
 
 const brTrainData = trainData.map(d => ({input: d, output: d}));
 const brSingleData = brTrainData.slice(0, 1);

diff --git a/examples/src/benchmark_gpu.js b/examples/src/benchmark_gpu.js
@@ -47,7 +47,7 @@ tfModel.compile({loss: "meanSquaredError", optimizer: "sgd"});
 
 const trainData = Matrix.random_2d(Count, Sizes[0]);
 
-const tfTrainData = tf.tensor(Matrix.copy_2d(trainData));
+const tfTrainData = tf.tensor(trainData.map(t => Array.from(t)));
 
 const brTrainData = trainData.map(d => ({input: d, output: d}));