Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-12-24 07:14:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Migrate model class to Tensor

Browse Source
This commit is contained in:
Matúš Námešný
2025-11-22 22:40:38 +01:00
parent ca44ea4436
commit 51bcee01ab
11 changed files with 207 additions and 254 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
include/module.hpp
View File

@@ -19,14 +19,14 @@ class Module {
size_t output_size();
void register_layer(const std::string& name, Layer& layer);
void register_layer(const std::string& name, Layer* layer);
void register_module(Module& module);
const std::vector<std::pair<std::string, Layer&>>& get_layers() const;
const std::vector<std::pair<std::string, Layer*>>& get_layers() const;
protected:
std::vector<std::pair<std::string, Layer&>> layers;
std::vector<std::pair<std::string, Layer*>> layers;
CUDANet::Shape in_shape;
CUDANet::Shape out_shape;