| 15 | struct GeluConfig { |
| 16 | GeluApproximation approximation = GeluApproximation::ExactErf; |
| 17 | }; |
| 18 | |
| 19 | class ReluModule { |
| 20 | public: |
| 21 | const core::ModuleSchema & schema() const noexcept; |
| 22 | core::TensorValue build(core::ModuleBuildContext & ctx, const core::TensorValue & input) const; |
| 23 | static const core::ModuleSchema & static_schema() noexcept; |
| 24 | }; |
| 25 |
no outgoing calls
no test coverage detected