Function prod

tfjs-backend-webgpu/src/kernels/Prod.ts:23–31

(
    args: {inputs: ProdInputs, backend: WebGPUBackend, attrs: ProdAttrs})

import {reduce} from '../kernel_utils/reduce';

export function prod(
    args: {inputs: ProdInputs, backend: WebGPUBackend, attrs: ProdAttrs}):
    TensorInfo {
  const {inputs, backend, attrs} = args;
  const {x} = inputs;
  const {axis, keepDims} = attrs;

  return reduce(x, axis, keepDims, 'prod', backend);
}

export const prodConfig: KernelConfig = {
  kernelName: Prod,