From 313d6278fbd5607be851a65a4ee261940707382a Mon Sep 17 00:00:00 2001
From: Daniel Smilkov <dsmilkov@gmail.com>
Date: Mon, 9 Dec 2019 10:55:43 -0500
Subject: [PATCH 1/3] save

---
 tfjs-backend-wasm/src/setup_test.ts | 1 +
 tfjs-core/src/ops/unary_ops.ts      | 7 +++++--
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/tfjs-backend-wasm/src/setup_test.ts b/tfjs-backend-wasm/src/setup_test.ts
index 86c9f34faf5..bcc0d59483e 100644
--- a/tfjs-backend-wasm/src/setup_test.ts
+++ b/tfjs-backend-wasm/src/setup_test.ts
@@ -193,6 +193,7 @@ const TEST_FILTERS: TestFilter[] = [
   {include: 'addN'},
   {include: 'nonMaxSuppression'},
   {include: 'argmax', excludes: ['gradient']},
+  {include: 'exp '},
 ];
 
 const customInclude = (testName: string) => {
diff --git a/tfjs-core/src/ops/unary_ops.ts b/tfjs-core/src/ops/unary_ops.ts
index 55977225d40..3586397bc61 100644
--- a/tfjs-core/src/ops/unary_ops.ts
+++ b/tfjs-core/src/ops/unary_ops.ts
@@ -212,13 +212,16 @@ function exp_<T extends Tensor>(x: T|TensorLike): T {
   const $x = convertToTensor(x, 'x', 'exp');
 
   const bck = (dy: T, saved: Tensor[]) => {
-    return {$x: () => dy.mulStrict(saved[0] as T)};
+    return {x: () => dy.mulStrict(saved[0] as T)};
   };
+  const attrs = {};
+  const inputsToSave: Tensor[] = [];
+  const outputsToSave = [true];
   return ENGINE.runKernelFunc((backend, save) => {
     const y = backend.exp($x);
     save([y]);
     return y;
-  }, {$x}, bck);
+  }, {x: $x}, bck, 'Exp', attrs, inputsToSave, outputsToSave);
 }
 
 /**

From e91a7eec06934b5357164727d19ce85617ff8cb7 Mon Sep 17 00:00:00 2001
From: Daniel Smilkov <dsmilkov@gmail.com>
Date: Mon, 9 Dec 2019 11:25:02 -0500
Subject: [PATCH 2/3] save

---
 tfjs-backend-wasm/src/kernels/all_kernels.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tfjs-backend-wasm/src/kernels/all_kernels.ts b/tfjs-backend-wasm/src/kernels/all_kernels.ts
index 363b27b8e28..51819c27127 100644
--- a/tfjs-backend-wasm/src/kernels/all_kernels.ts
+++ b/tfjs-backend-wasm/src/kernels/all_kernels.ts
@@ -31,6 +31,7 @@ import './Conv2D';
 import './CropAndResize';
 import './DepthwiseConv2dNative';
 import './Div';
+import './Exp';
 import './FloorDiv';
 import './FusedBatchNorm';
 import './FusedConv2D';
@@ -42,13 +43,12 @@ import './Mul';
 import './NonMaxSuppressionV3';
 import './PadV2';
 import './Prelu';
-import './Reshape';
 import './Relu';
 import './Relu6';
+import './Reshape';
 import './ResizeBilinear';
 import './Sigmoid';
 import './Slice';
 import './Square';
 import './Sub';
 import './Transpose';
-import './Exp';

From 78ac4cd01d0ecfdc144aa5b35afb94151c951241 Mon Sep 17 00:00:00 2001
From: Daniel Smilkov <dsmilkov@gmail.com>
Date: Mon, 9 Dec 2019 13:05:06 -0500
Subject: [PATCH 3/3] save

---
 tfjs-backend-wasm/scripts/test-bundle-size.js | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tfjs-backend-wasm/scripts/test-bundle-size.js b/tfjs-backend-wasm/scripts/test-bundle-size.js
index da511c2174a..0d9f2727a42 100755
--- a/tfjs-backend-wasm/scripts/test-bundle-size.js
+++ b/tfjs-backend-wasm/scripts/test-bundle-size.js
@@ -36,7 +36,9 @@ exec(
 
 shell.cd(dirName);
 shell.cd(wasmDirName);
-exec(`yarn && ./scripts/build-ci.sh && yarn rollup -c`, {silent: false});
+exec(
+    `yarn && yarn build-core && ./scripts/build-ci.sh && yarn rollup -c`,
+    {silent: false});
 
 const masterMinBundleSize = getFileSizeBytes(bundleFilename);
 const masterWasmSize = getFileSizeBytes(wasmFileName);