mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-07 02:04:26 +00:00
Rename dim2d to shape2d
This commit is contained in:
@@ -10,7 +10,7 @@
|
||||
using namespace CUDANet::Layers;
|
||||
|
||||
BatchNorm2d::BatchNorm2d(
|
||||
dim2d inputSize,
|
||||
shape2d inputSize,
|
||||
int inputChannels,
|
||||
float epsilon,
|
||||
ActivationType activationType
|
||||
@@ -128,7 +128,7 @@ int BatchNorm2d::getOutputSize() {
|
||||
return inputSize.first * inputSize.second * inputChannels;
|
||||
}
|
||||
|
||||
dim2d BatchNorm2d::getOutputDims() {
|
||||
shape2d BatchNorm2d::getOutputDims() {
|
||||
return inputSize;
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user