Expand description
PReLU FW plan — Milestone 5.3.
y[..., c, ...] = x[..., c, ...] if positive, else
weight[c] * x[..., c, ...].
weight is either per-channel (shape == [C], with C the size of the
channel axis) or a single learnable scalar (shape == [1]).
Distinct from crate::UnaryParamPlan because the parameter is a tensor
operand, not a scalar — needs its own plan shape.
Structs§
- PRelu
Args - Args bundle for a PReLU FW launch.
- PRelu
Descriptor - Descriptor for a PReLU FW op.
- PRelu
Plan - PReLU forward plan.