| 133 | "Snake1d", |
| 134 | "nn.activation", |
| 135 | kSnakeInputs, |
| 136 | 2, |
| 137 | kActivationOutputs, |
| 138 | 1, |
| 139 | "Applies Snake activation over channel-time tensors using per-channel alpha.", |
| 140 | }; |
| 141 | |
| 142 | template <typename Fn> |
| 143 | core::TensorValue build_unary( |
| 144 | core::ModuleBuildContext & ctx, |
| 145 | const core::TensorValue & input, |
| 146 | Fn fn) { |