Method getInCoord

tfjs-backend-webgl/src/reverse_gpu.ts:43–48

(i: number)

      return;
    }
    const getInCoord = (i: number) => {
      if (axis.indexOf(i) !== -1 && xShape[i] !== 1) {
        return `${xShape[i]} - coords[${i}] - 1`;
      }
      return `coords[${i}]`;
    };
    const inCoords = xShape.map((_, i) => getInCoord(i)).join(',');
    const type = getCoordsDataType(rank);