| 66 | }; |
| 67 | |
| 68 | class ConcatModule { |
| 69 | public: |
| 70 | explicit ConcatModule(ConcatConfig config); |
| 71 | |
| 72 | const ConcatConfig & config() const noexcept; |
| 73 | const core::ModuleSchema & schema() const noexcept; |
| 74 | core::TensorValue build( |
| 75 | core::ModuleBuildContext & ctx, |
| 76 | const core::TensorValue & lhs, |
| 77 | const core::TensorValue & rhs) const; |
| 78 | static const core::ModuleSchema & static_schema() noexcept; |
| 79 | |
| 80 | private: |
| 81 | ConcatConfig config_; |
| 82 | }; |
| 83 | |
| 84 | struct RepeatConfig { |
| 85 | core::TensorShape output_shape; |
no outgoing calls
no test coverage detected