Function eluGrad

tfjs-backend-webgpu/src/kernels/EluGrad.ts:25–32

(args: {inputs: EluGradInputs, backend: WebGPUBackend})

export const eluGrad =
    (args: {inputs: EluGradInputs, backend: WebGPUBackend}): TensorInfo => {
      const {inputs, backend} = args;
      const {dy, y} = inputs;

      const program =
          new BinaryOpProgram(BinaryOpType.ELU_DER, dy.shape, y.shape);
      return backend.runWebGPUProgram(program, [dy, y], dy.dtype);
    };

export const eluGradConfig: KernelConfig = {
  kernelName: EluGrad,