Format

Author: syt123450

src/ops/pool.ts

@@ -532,7 +532,7 @@ function avgPool3d_(
     dimRoundingMode?: 'floor'|'round'|'ceil',
     dataFormat: 'NDHWC'|'NCDHW' = 'NDHWC'): Tensor5D {
   const $x = convertToTensor(x, 'x', 'avgPool3d', 'float32');
-  
+
   if (dilations == null) {
     dilations = [1, 1, 1];
   }
@@ -553,22 +553,22 @@ function avgPool3d_(
         () => `Error in avgPool3d: pad must be an integer when using, ` +
             `dimRoundingMode ${dimRoundingMode} but got pad ${pad}.`);
   }
-  
+
   const convInfo = conv_util.computePool3DInfo(
       $x.shape, filterSize, strides, dilations, pad, dimRoundingMode,
       dataFormat);
-  
+
   const grad = (dy: Tensor5D) => {
     return {
       x: () => avgPool3dBackprop(
           dy, $x, filterSize, strides, dilations, pad, dimRoundingMode)
     };
   };
-  
+
   let res = ENGINE.runKernel(
       backend => backend.avgPool3d($x, convInfo), {x: $x}, grad);
   res = res.cast($x.dtype);
-  
+
   return res;
 }
 
@@ -691,7 +691,7 @@ function maxPool3d_(
     dimRoundingMode?: 'floor'|'round'|'ceil',
     dataFormat: 'NDHWC'|'NCDHW' = 'NDHWC'): Tensor5D {
   const $x = convertToTensor(x, 'x', 'maxPool3d');
-  
+
   if (dilations == null) {
     dilations = [1, 1, 1];
   }
@@ -712,11 +712,11 @@ function maxPool3d_(
         () => `Error in maxPool3d: pad must be an integer when using, ` +
             `dimRoundingMode ${dimRoundingMode} but got pad ${pad}.`);
   }
-  
+
   const convInfo = conv_util.computePool3DInfo(
       $x.shape, filterSize, strides, dilations, pad, dimRoundingMode,
       dataFormat);
-  
+
   const grad = (dy: Tensor5D, saved: Tensor[]) => {
     const [$x, y] = saved;
     return {
@@ -725,13 +725,13 @@ function maxPool3d_(
           pad, dimRoundingMode)
     };
   };
-  
+
   const res = ENGINE.runKernel((backend, save) => {
     const y = backend.maxPool3d($x, convInfo);
     save([$x, y]);
     return y;
   }, {x: $x}, grad);
-  
+
   return res;
 }