mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-12-23 23:04:25 +00:00
Refactor size calculations in layers and backend
This commit is contained in:
@@ -15,10 +15,6 @@ class Module {
|
||||
|
||||
CUDANet::Shape output_shape();
|
||||
|
||||
size_t input_size();
|
||||
|
||||
size_t output_size();
|
||||
|
||||
void register_layer(const std::string& name, Layer* layer);
|
||||
|
||||
void register_module(Module& module);
|
||||
|
||||
Reference in New Issue
Block a user