Function kernelFunc

tfjs-node/src/kernels/FlipLeftRight.ts:25–39

({inputs, backend})

  kernelName: FlipLeftRight,
  backendName: 'tensorflow',
  kernelFunc({inputs, backend}) {
    const nodeBackend = backend as NodeJSKernelBackend;
    const {image} = inputs as FlipLeftRightInputs;
    const opAttrs = [
      createTensorsTypeOpAttr('Tidx', 'int32'),
      createTensorsTypeOpAttr('T', image.dtype),
    ];
    const axes = util.parseAxisParam([2], image.shape);
    const axisTensor = tensor1d(axes, 'int32');
    const res =
        nodeBackend.executeSingleOutput(
            'ReverseV2', opAttrs, [image, axisTensor]);
    axisTensor.dispose();
    return res;
  }
};