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

Rename dim2d to shape2d

Browse Source
This commit is contained in:
Matúš Námešný
2024-05-27 21:14:51 +02:00
parent 07d505a0e5
commit df47a31f36
22 changed files with 120 additions and 120 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
include/layers/batch_norm.cuh
View File

@@ -10,7 +10,7 @@ namespace CUDANet::Layers {
class BatchNorm2d : public WeightedLayer, public TwoDLayer {
public:
BatchNorm2d(dim2d inputSize, int inputChannels, float epsilon, ActivationType activationType);
BatchNorm2d(shape2d inputSize, int inputChannels, float epsilon, ActivationType activationType);
~BatchNorm2d();
@@ -64,11 +64,11 @@ class BatchNorm2d : public WeightedLayer, public TwoDLayer {
*/
int getInputSize();
dim2d getOutputDims();
shape2d getOutputDims();
private:
dim2d inputSize;
shape2d inputSize;
int inputChannels;
int gridSize;