| 77 | }; |
| 78 | |
| 79 | class GLUModule { |
| 80 | public: |
| 81 | const core::ModuleSchema & schema() const noexcept; |
| 82 | core::TensorValue build(core::ModuleBuildContext & ctx, const core::TensorValue & input) const; |
| 83 | static const core::ModuleSchema & static_schema() noexcept; |
| 84 | }; |
| 85 | |
| 86 | struct Snake1dConfig { |
| 87 | int64_t hidden_size = 0; |