Add this.throwIfDisposed();

tfjs-core/src/public/chained_ops/add.ts

@@ -25,5 +25,6 @@ declare module '../../tensor' {
 }
 
 Tensor.prototype.add = function<T extends Tensor>(b: Tensor|TensorLike): T {
+  this.throwIfDisposed();
   return add(this, b);
 };

tfjs-core/src/public/chained_ops/batchnorm.ts

@@ -21,20 +21,20 @@ import {Rank, TensorLike} from '../../types';
 declare module '../../tensor' {
   interface Tensor<R extends Rank = Rank> {
     batchNorm<T extends Tensor>(
-      mean: Tensor<R> | Tensor1D | TensorLike,
-      variance: Tensor<R> | Tensor1D | TensorLike,
-      offset?: Tensor<R> | Tensor1D | TensorLike,
-      scale?: Tensor<R> | Tensor1D | TensorLike,
-      varianceEpsilon?: number
-    ): Tensor<R>;
+        mean: Tensor<R>|Tensor1D|TensorLike,
+        variance: Tensor<R>|Tensor1D|TensorLike,
+        offset?: Tensor<R>|Tensor1D|TensorLike,
+        scale?: Tensor<R>|Tensor1D|TensorLike,
+        varianceEpsilon?: number): Tensor<R>;
   }
 }
 
-Tensor.prototype.batchNorm = function <R extends Rank>(
-  this: Tensor<R> | TensorLike, mean: Tensor<R> | Tensor1D | TensorLike,
-  variance: Tensor<R> | Tensor1D | TensorLike,
-  offset?: Tensor<R> | Tensor1D | TensorLike,
-  scale?: Tensor<R> | Tensor1D | TensorLike,
-  varianceEpsilon?: number): Tensor<R> {
+Tensor.prototype.batchNorm = function<R extends Rank>(
+    mean: Tensor<R>|Tensor1D|TensorLike,
+    variance: Tensor<R>|Tensor1D|TensorLike,
+    offset?: Tensor<R>|Tensor1D|TensorLike,
+    scale?: Tensor<R>|Tensor1D|TensorLike,
+    varianceEpsilon?: number): Tensor<R> {
+  this.throwIfDisposed();
   return batchNorm(this, mean, variance, offset, scale, varianceEpsilon);
 };

tfjs-core/src/public/chained_ops/broadcast_to.ts

@@ -27,5 +27,6 @@ declare module '../../tensor' {
 
 Tensor.prototype.broadcastTo = function<R extends Rank>(shape: ShapeMap[R]):
     Tensor<R> {
+  this.throwIfDisposed();
   return broadcastTo(this, shape);
 };

tfjs-core/src/public/chained_ops/div.ts

@@ -26,5 +26,6 @@ declare module '../../tensor' {
 }
 
 Tensor.prototype.div = function<T extends Tensor>(b: Tensor|TensorLike): T {
+  this.throwIfDisposed();
   return div(this, b);
 };

tfjs-core/src/public/chained_ops/div_no_nan.ts

@@ -27,5 +27,6 @@ declare module '../../tensor' {
 
 Tensor.prototype.divNoNan = function<T extends Tensor>(b: Tensor|
                                                        TensorLike): T {
+  this.throwIfDisposed();
   return divNoNan(this, b);
 };

tfjs-core/src/public/chained_ops/one_hot.ts

@@ -27,5 +27,6 @@ declare module '../../tensor' {
 
 Tensor.prototype.oneHot = function(
     depth: number, onValue = 1, offValue = 0): Tensor {
+  this.throwIfDisposed();
   return oneHot(this, depth, onValue, offValue);
 };

tfjs-core/src/public/chained_ops/pad.ts

@@ -28,5 +28,6 @@ declare module '../../tensor' {
 
 Tensor.prototype.pad = function<T extends Tensor>(
     this: T, paddings: Array<[number, number]>, constantValue: number): T {
+  this.throwIfDisposed();
   return pad(this, paddings, constantValue);
 };

tfjs-core/src/public/chained_ops/squared_difference.ts

@@ -27,5 +27,6 @@ declare module '../../tensor' {
 
 Tensor.prototype.squaredDifference = function<T extends Tensor>(b: Tensor|
                                                                 TensorLike): T {
+  this.throwIfDisposed();
   return squaredDifference(this, b);
 };

tfjs-core/src/public/chained_ops/tile.ts

@@ -26,5 +26,6 @@ declare module '../../tensor' {
 }
 
 Tensor.prototype.tile = function<T extends Tensor>(reps: number[]): T {
+  this.throwIfDisposed();
   return tile(this, reps) as T;
 };

tfjs-core/src/public/chained_ops/transpose.ts

@@ -27,5 +27,6 @@ declare module '../../tensor' {
 
 Tensor.prototype.transpose = function<T extends Tensor>(
     this: T, perm?: number[]): T {
+  this.throwIfDisposed();
   return transpose(this, perm);
 };