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

Refactor size calculations in layers and backend

Browse Source
This commit is contained in:
Matúš Námešný
2025-11-27 22:01:09 +01:00
parent c855ae89ec
commit e79667671a
13 changed files with 58 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/layers/avg_pooling.cpp
View File

@@ -84,11 +84,11 @@ CUDANet::Shape AvgPool2d::output_shape() {
}
size_t AvgPool2d::input_size() {
return sizeof(float) * in_shape[0] * in_shape[1] * in_shape[2];
return dtype_size(dtype) * in_shape[0] * in_shape[1] * in_shape[2];
}
size_t AvgPool2d::output_size() {
return sizeof(float) * out_shape[0] * out_shape[1] * out_shape[2];
return dtype_size(dtype) * out_shape[0] * out_shape[1] * out_shape[2];
}
void AvgPool2d::set_weights(void* input) {}