| 157 | }; |
| 158 | |
| 159 | class MaskingModule { |
| 160 | public: |
| 161 | const core::ModuleSchema & schema() const noexcept; |
| 162 | core::TensorValue build( |
| 163 | core::ModuleBuildContext & ctx, |
| 164 | const core::TensorValue & input, |
| 165 | const core::TensorValue & mask) const; |
| 166 | static const core::ModuleSchema & static_schema() noexcept; |
| 167 | }; |
| 168 | |
| 169 | } // namespace engine::modules |
no outgoing calls
no test coverage detected