Method constructor

tfjs-backend-webgl/src/conv_packed_gpu.ts:36–410

(
       convInfo: backend_util.Conv2DInfo, addBias = false,
       activation: string = null, hasPreluActivation = false,
       hasLeakyReluAlpha = false)

   ];

   constructor(
       convInfo: backend_util.Conv2DInfo, addBias = false,
       activation: string = null, hasPreluActivation = false,
       hasLeakyReluAlpha = false) {
     this.outputShape = convInfo.outShape;
     this.enableShapeUniforms = useShapeUniforms(this.outputShape.length);
     const padLeft = convInfo.padInfo.left;
     const strideWidth = convInfo.strideWidth;
     const dilationWidth = convInfo.dilationWidth;
     const filterHeight = convInfo.filterHeight;
     const filterWidth = convInfo.filterWidth;
     const texelsAcross = filterWidth;

     let mainLoop = `
       int xR; int xC; int xCOffset;
       vec4 wTexel; vec4 previous; vec4 final;`;

     for (let c = 0; c < filterWidth; c++) {
       mainLoop += `
           vec4 xTexelC${c * 2};
           int xTexelC${c * 2}Ready;
           vec4 xTexelC${c * 2 + 1};
           int xTexelC${c * 2 + 1}Ready;
           vec4 xC${c};`;
     }

     /**
      * This vectorized implementation works by gathering the values needed for
      * each output channel's dot product into vec4's and then multiplying them
      * all together (this happens in the final double for-loop below). Most of
      * the main loop consists of constructing these vec4's with the minimum
      * number of texture2D calls, which means making use of all four returned
      * values from a texture2D call at once.
      */
     mainLoop += `
     for (int r = 0; r < ${filterHeight}; r++) {
      for (int d1 = 0; d1 < ${convInfo.inChannels}; d1 += 2) {
       `;
     for (let c = 0; c < filterWidth; c++) {
       mainLoop += `
           xTexelC${c * 2} = vec4(0.0);
           xTexelC${c * 2}Ready = 0;
           xTexelC${c * 2 + 1} = vec4(0.0);
           xTexelC${c * 2 + 1}Ready = 0;
           xC${c} = vec4(0.0);`;
     }
     mainLoop += `
         xR = xRCorner + r * dilations[0];
         if (xR >=0 && xR < inDims[0]) {
       `;

     for (let texelC = 0; texelC < (texelsAcross + 1) / 2; texelC++) {
       const colIndex = texelC * 2;

       mainLoop += `
           xC = xCCorner + ${colIndex * dilationWidth};
           `;