| 121 | 1, |
| 122 | kActivationOutputs, |
| 123 | 1, |
| 124 | "Applies ELU activation elementwise.", |
| 125 | }; |
| 126 | |
| 127 | const core::ModuleSchema kSoftmaxSchema = { |
| 128 | "Softmax", |
| 129 | "nn.activation", |
| 130 | kActivationInputs, |
| 131 | 1, |
| 132 | kActivationOutputs, |
| 133 | 1, |
| 134 | "Applies softmax over the last physical dimension.", |
| 135 | }; |
| 136 |
no test coverage detected