[node] Use makeTensorFromTensorInfo instead of makeTensorFromId (#6361)

* [node] Use makeTensorFromTensorInfo instead of makeTensorFromId See #6355 * [node] Disable upperBound and lowerBound tests Co-authored-by: Na Li <linazhao@google.com>
tensorflow · May 3, 2022 · ba3a4cb · ba3a4cb
1 parent 90b4288
commit ba3a4cb
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/tfjs-node/src/nodejs_kernel_backend.ts b/tfjs-node/src/nodejs_kernel_backend.ts
@@ -120,7 +120,9 @@ export class NodeJSKernelBackend extends KernelBackend {
     // We can then change the return type from Tensor to TensorInfo.
     // return {dataId: newId, shape: metadata.shape, dtype};
 
-    return tf.engine().makeTensorFromDataId(newId, metadata.shape, dtype);
+    const tensorInfo: TensorInfo = {
+      dataId: newId, shape: metadata.shape, dtype};
+    return tf.engine().makeTensorFromTensorInfo(tensorInfo);
   }
 
   // Prepares Tensor instances for Op execution.

diff --git a/tfjs-node/src/run_tests.ts b/tfjs-node/src/run_tests.ts
@@ -162,6 +162,9 @@ const IGNORE_LIST: string[] = [
   // int32 input for the following ops not supported by TF 2.4.0.
   'leakyrelu test-tensorflow {} int32',
   'step kernel test-tensorflow {} with int32 tensor',
+  // upperBound and lowerBound use SearchSorted, which is unsupported
+  'upperBound',
+  'lowerBound',
 ];
 
 if (process.platform === 'win32') {