mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-12-23 14:54:28 +00:00
Refactor Layer interface to return size of weights and biases instead of Tensor references
This commit is contained in:
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user