Python wrappers
Public Python APIs implemented by this engine symbol.Autograd node for ReLU6 (Krizhevsky 2010, MobileNet family).
Gradient is the indicator of the open interval — built directly as an in-range storage mask and multiplied with the upstream gradient, avoiding a backend round-trip for this very simple piecewise rule.
Math
\qquad \frac{\partial y}{\partial x} = \begin{cases} 1, & 0 < x < 6 \\ 0, & \text{otherwise} \end{cases}$$Attributes
schema_v1OpSchema"relu6", AmpPolicy::KeepInput.