| 159 | } |
| 160 | |
| 161 | Encoder::Encoder(const VisionConfig &Config) { |
| 162 | for (int I = 0; I < Config.NumHiddenLayers; I++) { |
| 163 | Layers.push_back(std::make_shared<EncoderLayer>(Config)); |
| 164 | } |
| 165 | registerLayer("layers", Layers); |
| 166 | } |
| 167 | |
| 168 | std::pair<mx::array, std::vector<mx::array>> |
| 169 | Encoder::forward(const mx::array &X, |
nothing calls this directly
no outgoing calls
no test coverage detected