Method constructor

tfjs-backend-webgl/src/transform_gpu.ts:25–164

(
      imageHeight: number, imageWidth: number,
      interpolation: 'nearest'|'bilinear',
      fillMode: 'constant'|'reflect'|'wrap'|'nearest', fillValue: number,
      outShape: [number, number, number, number])

  userCode: string;

  constructor(
      imageHeight: number, imageWidth: number,
      interpolation: 'nearest'|'bilinear',
      fillMode: 'constant'|'reflect'|'wrap'|'nearest', fillValue: number,
      outShape: [number, number, number, number]) {
    this.outputShape = outShape;
    const interpolationModeId = interpolation === 'nearest' ? 1 : 2;
    let fillModeId;
    switch (fillMode) {
      case 'constant':
        fillModeId = 1;
        break;
      case 'reflect':
        fillModeId = 2;
        break;
      case 'wrap':
        fillModeId = 3;
        break;
      case 'nearest':
        fillModeId = 4;
        break;
      default:
        fillModeId = 1;
        break;
    }
    this.userCode = `
            float mapCoord(float outCoord, float len) {
              float inCoord = outCoord;
              if(${fillModeId} == 2) {
                if (inCoord < 0.0) {
                  if (len <= 1.0) {
                    inCoord = 0.0;
                  } else {
                    float sz2 = 2.0 * len;
                    if (inCoord < sz2) {
                      inCoord = sz2 * float(int(float(-inCoord / sz2))) +
                      inCoord;
                    }
                    inCoord = inCoord < -len ? inCoord + sz2 : -inCoord - 1.0;
                  }
                } else if (inCoord > len - 1.0) {
                  if (len <= 1.0) {
                    inCoord = 0.0;
                  } else {
                    float sz2 = 2.0 * len;
                    inCoord -= sz2 * float(int(float(inCoord / sz2)));
                    if (inCoord >= len) {
                      inCoord = sz2 - inCoord - 1.0;
                    }
                  }
                }
                return clamp(inCoord, 0.0, len - 1.0);
              } else if (${fillModeId} == 3) {
                if (inCoord < 0.0) {
                  if (len <= 1.0) {
                    inCoord = 0.0;
                  } else {
                    float sz = len - 1.0;