|
|
|
@ -150,18 +150,12 @@ void getPoolingKernelParams(const LayerParams ¶ms, std::vector<size_t>& kern |
|
|
|
|
{ |
|
|
|
|
bool is_global = params.get<bool>("global_pooling", false); |
|
|
|
|
globalPooling = std::vector<bool>(3, is_global); |
|
|
|
|
if (params.has("global_d")) |
|
|
|
|
{ |
|
|
|
|
globalPooling[0] = params.get<bool>("global_d"); |
|
|
|
|
} |
|
|
|
|
else if (params.has("global_h")) |
|
|
|
|
{ |
|
|
|
|
globalPooling[1] = params.get<bool>("global_h"); |
|
|
|
|
} |
|
|
|
|
else if (params.has("global_w")) |
|
|
|
|
{ |
|
|
|
|
globalPooling[2] = params.get<bool>("global_w"); |
|
|
|
|
} |
|
|
|
|
if (params.has("global_pooling_d")) |
|
|
|
|
globalPooling[0] = params.get<bool>("global_pooling_d"); |
|
|
|
|
else if (params.has("global_pooling_h")) |
|
|
|
|
globalPooling[1] = params.get<bool>("global_pooling_h"); |
|
|
|
|
else if (params.has("global_pooling_w")) |
|
|
|
|
globalPooling[2] = params.get<bool>("global_pooling_w"); |
|
|
|
|
|
|
|
|
|
if (is_global) |
|
|
|
|
{ |
|
|
|
|