torch.CPUKernelConfig
export interface CPUKernelConfig {
/**
* Forward function implementing the operation.
* Signature depends on operation kind:
* - Unary: (input, out, numel) => void
* - Binary: (a, b, out, numel) => void
* - Reduction: (input, out, numel, dim, keepdim) => void
*/
readonly forward: CPUForwardFn;
}- readonly
forward(CPUForwardFn) - – Forward function implementing the operation. Signature depends on operation kind: - Unary: (input, out, numel) => void - Binary: (a, b, out, numel) => void - Reduction: (input, out, numel, dim, keepdim) => void
CPU kernel configuration.