torch.AutogradEntry

export interface AutogradEntry {
  readonly op: string;
  readonly dtype?: AutogradDType;
  readonly device: AutogradDevice;
  readonly setup_context?: (
    ctx: GradContext,
    inputs: unknown,
    output: unknown,
    meta?: unknown
  ) => void;
  readonly backward: (ctx: GradContext, grad_output: unknown) => unknown;
}

readonlyop(string)
readonlydtype(AutogradDType)optional
readonlydevice(AutogradDevice)
readonlysetup_context(( ctx: GradContext, inputs: unknown, output: unknown, meta?: unknown ) => void)optional
readonlybackward((ctx: GradContext, grad_output: unknown) => unknown)

Internal autograd entry stored in registry.