Function transform

tfjs-backend-webgl/src/kernels/Transform.ts:23–42

(args: {
  inputs: TransformInputs,
  backend: MathBackendWebGL,
  attrs: TransformAttrs
})

import {TransformProgram} from '../transform_gpu';

export function transform(args: {
  inputs: TransformInputs,
  backend: MathBackendWebGL,
  attrs: TransformAttrs
}): TensorInfo {
  const {inputs, backend, attrs} = args;
  const {image, transforms} = inputs;
  const {interpolation, fillMode, fillValue, outputShape} = attrs;

  const [batch, imageHeight, imageWidth, numChannels] = image.shape;
  const [outHeight, outWidth] =
      outputShape != null ? outputShape : [imageHeight, imageWidth];
  const outShape =
      [batch, outHeight, outWidth,
       numChannels] as [number, number, number, number];

  const program = new TransformProgram(
      imageHeight, imageWidth, interpolation, fillMode, fillValue, outShape);
  return backend.runWebGLProgram(program, [image, transforms], 'float32');
}

export const transformConfig: KernelConfig = {
  kernelName: Transform,