[node] Remove deprecated createTypeOpAttr (#3235)

BREAKING remove deprecated function createTypeOpAttr
tensorflow · May 11, 2020 · 46ca8d7 · 46ca8d7
1 parent 138f23a
commit 46ca8d7
Show file tree

Hide file tree

Showing 5 changed files with 153 additions and 141 deletions.
diff --git a/tfjs-node/src/kernels/Softmax.ts b/tfjs-node/src/kernels/Softmax.ts
@@ -17,7 +17,7 @@
 
 import {NamedTensorInfoMap, registerKernel, TensorInfo} from '@tensorflow/tfjs';
 
-import {createTypeOpAttr, NodeJSKernelBackend} from '../nodejs_kernel_backend';
+import {createTensorsTypeOpAttr, NodeJSKernelBackend} from '../nodejs_kernel_backend';
 
 interface SoftmaxInputs extends NamedTensorInfoMap {
   logits: TensorInfo;
@@ -28,7 +28,7 @@ registerKernel({
   backendName: 'tensorflow',
   kernelFunc: ({inputs, backend}) => {
     const {logits} = inputs as SoftmaxInputs;
-    const opAttrs = [createTypeOpAttr('T', logits.dtype)];
+    const opAttrs = [createTensorsTypeOpAttr('T', logits.dtype)];
 
     const nodeBackend = backend as NodeJSKernelBackend;
 

diff --git a/tfjs-node/src/kernels/SquaredDifference.ts b/tfjs-node/src/kernels/SquaredDifference.ts
@@ -16,7 +16,7 @@
  */
 
 import {KernelConfig, registerKernel} from '@tensorflow/tfjs';
-import {createTypeOpAttr, NodeJSKernelBackend} from '../nodejs_kernel_backend';
+import {createTensorsTypeOpAttr, NodeJSKernelBackend} from '../nodejs_kernel_backend';
 
 export const squaredDifferenceConfig: KernelConfig = {
   // TODO import this kernelName from core once exported.
@@ -25,7 +25,7 @@ export const squaredDifferenceConfig: KernelConfig = {
   kernelFunc: ({inputs, backend}) => {
     const {a, b} = inputs;
 
-    const opAttrs = [createTypeOpAttr('T', a.dtype)];
+    const opAttrs = [createTensorsTypeOpAttr('T', a.dtype)];
     const nodeBackend = backend as NodeJSKernelBackend;
 
     return nodeBackend.executeSingleOutput(

diff --git a/tfjs-node/src/kernels/non_max_suppression_v5.ts b/tfjs-node/src/kernels/non_max_suppression_v5.ts
@@ -17,7 +17,7 @@
 
 import {NamedAttrMap, NamedTensorInfoMap, registerKernel, scalar, Tensor1D, Tensor2D, TensorInfo} from '@tensorflow/tfjs';
 
-import {createTypeOpAttr, NodeJSKernelBackend} from '../nodejs_kernel_backend';
+import {createTensorsTypeOpAttr, NodeJSKernelBackend} from '../nodejs_kernel_backend';
 
 interface NonMaxSuppressionWithScoreInputs extends NamedTensorInfoMap {
   boxes: TensorInfo;
@@ -43,7 +43,7 @@ registerKernel({
     const iouThresholdTensor = scalar(iouThreshold);
     const scoreThresholdTensor = scalar(scoreThreshold);
     const softNmsSigmaTensor = scalar(softNmsSigma);
-    const opAttrs = [createTypeOpAttr('T', boxes.dtype)];
+    const opAttrs = [createTensorsTypeOpAttr('T', boxes.dtype)];
 
     const nodeBackend = backend as NodeJSKernelBackend;