Refactor Layer interface to return size of weights and biases instead of Tensor references

This commit is contained in:
2025-11-23 20:44:25 +01:00
parent 547cd0c224
commit 9f1a56c699
14 changed files with 48 additions and 36 deletions

View File

@@ -81,11 +81,15 @@ size_t AvgPool2d::output_size() {
void AvgPool2d::set_weights(void* input) {}
CUDANet::Tensor& AvgPool2d::get_weights() {}
size_t AvgPool2d::get_weights_size() {
return 0;
}
void AvgPool2d::set_biases(void* input) {}
CUDANet::Tensor& AvgPool2d::get_biases() {}
size_t AvgPool2d::get_biases_size() {
return 0;
}
AdaptiveAvgPool2d::AdaptiveAvgPool2d(