| 63 | public: |
| 64 | explicit GeluModule(GeluConfig config = {}); |
| 65 | |
| 66 | const GeluConfig & config() const noexcept; |
| 67 | const core::ModuleSchema & schema() const noexcept; |
| 68 | core::TensorValue build(core::ModuleBuildContext & ctx, const core::TensorValue & input) const; |
| 69 | static const core::ModuleSchema & static_schema() noexcept; |
| 70 | |
| 71 | private: |
| 72 | GeluConfig config_; |
| 73 | }; |
no outgoing calls
no test coverage detected