diff --git a/tfjs-core/src/public/chained_ops/add.ts b/tfjs-core/src/public/chained_ops/add.ts index 83ca43ed49b..8ad4ac94640 100644 --- a/tfjs-core/src/public/chained_ops/add.ts +++ b/tfjs-core/src/public/chained_ops/add.ts @@ -25,5 +25,6 @@ declare module '../../tensor' { } Tensor.prototype.add = function(b: Tensor|TensorLike): T { + this.throwIfDisposed(); return add(this, b); }; diff --git a/tfjs-core/src/public/chained_ops/batchnorm.ts b/tfjs-core/src/public/chained_ops/batchnorm.ts index ce621c8eb04..84b3f1d67cd 100644 --- a/tfjs-core/src/public/chained_ops/batchnorm.ts +++ b/tfjs-core/src/public/chained_ops/batchnorm.ts @@ -21,20 +21,20 @@ import {Rank, TensorLike} from '../../types'; declare module '../../tensor' { interface Tensor { batchNorm( - mean: Tensor | Tensor1D | TensorLike, - variance: Tensor | Tensor1D | TensorLike, - offset?: Tensor | Tensor1D | TensorLike, - scale?: Tensor | Tensor1D | TensorLike, - varianceEpsilon?: number - ): Tensor; + mean: Tensor|Tensor1D|TensorLike, + variance: Tensor|Tensor1D|TensorLike, + offset?: Tensor|Tensor1D|TensorLike, + scale?: Tensor|Tensor1D|TensorLike, + varianceEpsilon?: number): Tensor; } } -Tensor.prototype.batchNorm = function ( - this: Tensor | TensorLike, mean: Tensor | Tensor1D | TensorLike, - variance: Tensor | Tensor1D | TensorLike, - offset?: Tensor | Tensor1D | TensorLike, - scale?: Tensor | Tensor1D | TensorLike, - varianceEpsilon?: number): Tensor { +Tensor.prototype.batchNorm = function( + mean: Tensor|Tensor1D|TensorLike, + variance: Tensor|Tensor1D|TensorLike, + offset?: Tensor|Tensor1D|TensorLike, + scale?: Tensor|Tensor1D|TensorLike, + varianceEpsilon?: number): Tensor { + this.throwIfDisposed(); return batchNorm(this, mean, variance, offset, scale, varianceEpsilon); }; diff --git a/tfjs-core/src/public/chained_ops/broadcast_to.ts b/tfjs-core/src/public/chained_ops/broadcast_to.ts index 49075b7df8a..187b17e20c8 100644 --- a/tfjs-core/src/public/chained_ops/broadcast_to.ts +++ b/tfjs-core/src/public/chained_ops/broadcast_to.ts @@ -27,5 +27,6 @@ declare module '../../tensor' { Tensor.prototype.broadcastTo = function(shape: ShapeMap[R]): Tensor { + this.throwIfDisposed(); return broadcastTo(this, shape); }; diff --git a/tfjs-core/src/public/chained_ops/div.ts b/tfjs-core/src/public/chained_ops/div.ts index 5491e286965..8d32570eda3 100644 --- a/tfjs-core/src/public/chained_ops/div.ts +++ b/tfjs-core/src/public/chained_ops/div.ts @@ -26,5 +26,6 @@ declare module '../../tensor' { } Tensor.prototype.div = function(b: Tensor|TensorLike): T { + this.throwIfDisposed(); return div(this, b); }; diff --git a/tfjs-core/src/public/chained_ops/div_no_nan.ts b/tfjs-core/src/public/chained_ops/div_no_nan.ts index ad09451be0f..89517f8c9f9 100644 --- a/tfjs-core/src/public/chained_ops/div_no_nan.ts +++ b/tfjs-core/src/public/chained_ops/div_no_nan.ts @@ -27,5 +27,6 @@ declare module '../../tensor' { Tensor.prototype.divNoNan = function(b: Tensor| TensorLike): T { + this.throwIfDisposed(); return divNoNan(this, b); }; diff --git a/tfjs-core/src/public/chained_ops/one_hot.ts b/tfjs-core/src/public/chained_ops/one_hot.ts index f6f6cf06f96..b3e523f3e74 100644 --- a/tfjs-core/src/public/chained_ops/one_hot.ts +++ 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); }; diff --git a/tfjs-core/src/public/chained_ops/pad.ts b/tfjs-core/src/public/chained_ops/pad.ts index 6be9d0037bb..deff71fe971 100644 --- a/tfjs-core/src/public/chained_ops/pad.ts +++ b/tfjs-core/src/public/chained_ops/pad.ts @@ -28,5 +28,6 @@ declare module '../../tensor' { Tensor.prototype.pad = function( this: T, paddings: Array<[number, number]>, constantValue: number): T { + this.throwIfDisposed(); return pad(this, paddings, constantValue); }; diff --git a/tfjs-core/src/public/chained_ops/squared_difference.ts b/tfjs-core/src/public/chained_ops/squared_difference.ts index 3e65837001f..d5a0c91249f 100644 --- a/tfjs-core/src/public/chained_ops/squared_difference.ts +++ b/tfjs-core/src/public/chained_ops/squared_difference.ts @@ -27,5 +27,6 @@ declare module '../../tensor' { Tensor.prototype.squaredDifference = function(b: Tensor| TensorLike): T { + this.throwIfDisposed(); return squaredDifference(this, b); }; diff --git a/tfjs-core/src/public/chained_ops/tile.ts b/tfjs-core/src/public/chained_ops/tile.ts index a586a3b8946..ebef41dad2a 100644 --- a/tfjs-core/src/public/chained_ops/tile.ts +++ b/tfjs-core/src/public/chained_ops/tile.ts @@ -26,5 +26,6 @@ declare module '../../tensor' { } Tensor.prototype.tile = function(reps: number[]): T { + this.throwIfDisposed(); return tile(this, reps) as T; }; diff --git a/tfjs-core/src/public/chained_ops/transpose.ts b/tfjs-core/src/public/chained_ops/transpose.ts index fc0bc73336c..42d1da54fd7 100644 --- a/tfjs-core/src/public/chained_ops/transpose.ts +++ b/tfjs-core/src/public/chained_ops/transpose.ts @@ -27,5 +27,6 @@ declare module '../../tensor' { Tensor.prototype.transpose = function( this: T, perm?: number[]): T { + this.throwIfDisposed(); return transpose(this, perm); };