Update to tfjs-core ~0.11.1 (tensorflow#93)

Author: nkreeger

package.json

@@ -20,7 +20,7 @@
     "publish-local": "yarn prep && yarn build && yalc push"
   },
   "devDependencies": {
-    "@tensorflow/tfjs-core": "~0.10.1",
+    "@tensorflow/tfjs-core": "~0.11.1",
     "@types/bindings": "~1.3.0",
     "@types/jasmine": "~2.8.6",
     "@types/node": "~9.6.2",
@@ -36,6 +36,6 @@
     "bindings": "~1.3.0"
   },
   "peerDependencies": {
-    "@tensorflow/tfjs-core": "~0.10.1"
+    "@tensorflow/tfjs-core": "~0.11.0"
   }
 }

src/nodejs_kernel_backend.ts

@@ -19,6 +19,7 @@
 import {BackendTimingInfo, DataType, fill, KernelBackend, ones, Rank, rsqrt, scalar, ShapeMap, Tensor, Tensor1D, tensor1d, Tensor2D, tensor2d, Tensor3D, Tensor4D} from '@tensorflow/tfjs-core';
 import {Conv2DInfo} from '@tensorflow/tfjs-core/dist/ops/conv_util';
 import {upcastType} from '@tensorflow/tfjs-core/dist/types';
+
 import {TensorMetadata, TFEOpAttr, TFJSBinding} from './tfjs_binding';
 
 type TensorInfo = {
@@ -190,6 +191,26 @@ export class NodeJSKernelBackend implements KernelBackend {
     return this.executeSingleOutput('MatMul', opAttrs, [a, b]) as Tensor2D;
   }
 
+  stridedSlice<T extends Tensor<Rank>>(
+      x: T, begin: number[], end: number[], strides: number[],
+      beginMask: number, endMask: number): T {
+    const beginTensor = tensor1d(begin, 'int32');
+    const endTensor = tensor1d(end, 'int32');
+    const stridesTensor = tensor1d(strides, 'int32');
+    const opAttrs = [
+      this.createTypeOpAttr('T', x.dtype),
+      this.createTypeOpAttr('Index', 'int32'),
+      {name: 'begin_mask', type: this.binding.TF_ATTR_INT, value: beginMask},
+      {name: 'end_mask', type: this.binding.TF_ATTR_INT, value: endMask},
+      {name: 'ellipsis_mask', type: this.binding.TF_ATTR_INT, value: 0},
+      {name: 'new_axis_mask', type: this.binding.TF_ATTR_INT, value: 0},
+      {name: 'shrink_axis_mask', type: this.binding.TF_ATTR_INT, value: 0}
+    ];
+    return this.executeSingleOutput(
+               'StridedSlice', opAttrs,
+               [x, beginTensor, endTensor, stridesTensor]) as T;
+  }
+
   slice<T extends Tensor>(x: T, begin: number[], size: number[]): T {
     const opAttrs = [
       this.createTypeOpAttr('T', x.dtype),
@@ -901,6 +922,18 @@ export class NodeJSKernelBackend implements KernelBackend {
     ]) as Tensor2D;
   }
 
+  cumsum(x: Tensor<Rank>, axis: number, exclusive: boolean, reverse: boolean):
+      Tensor<Rank> {
+    const axisTensor = scalar(axis, 'int32');
+    const opAttrs = [
+      {name: 'exclusive', type: this.binding.TF_ATTR_BOOL, value: exclusive},
+      {name: 'reverse', type: this.binding.TF_ATTR_BOOL, value: reverse},
+      this.createTypeOpAttr('T', x.dtype),
+      this.createTypeOpAttr('Tidx', 'int32')
+    ];
+    return this.executeSingleOutput('Cumsum', opAttrs, [x, axisTensor]);
+  }
+
   fromPixels(
       pixels: ImageData|HTMLImageElement|HTMLCanvasElement|HTMLVideoElement,
       numChannels: number): Tensor3D {

yarn.lock

@@ -2,9 +2,9 @@
 # yarn lockfile v1
 
 
-"@tensorflow/tfjs-core@~0.10.1":
-  version "0.10.1"
-  resolved "https://registry.yarnpkg.com/@tensorflow/tfjs-core/-/tfjs-core-0.10.1.tgz#1764c8c122d5b8a86a8d23c8ffd58bef8e40eda9"
+"@tensorflow/tfjs-core@~0.11.1":
+  version "0.11.1"
+  resolved "https://registry.yarnpkg.com/@tensorflow/tfjs-core/-/tfjs-core-0.11.1.tgz#d26808f912529668d0a41228da37566b6b2f4f08"
   dependencies:
     seedrandom "~2.4.3"