This commit is contained in:
Emad Barsoum 2016-11-09 19:50:32 -08:00
Родитель 8e5635585c
Коммит 0110d72f44
1 изменённых файлов: 4 добавлений и 0 удалений

Просмотреть файл

@ -547,6 +547,10 @@ namespace CNTK
// In case of pooling if the kernel shape is unknown, then treat it as global pooling.
if (kernelShape == NDShape::Unknown)
{
if ((std::find(autoPad.begin(), autoPad.end(), true) != autoPad.end()) ||
(lowerPad.TotalSize() > 0) || (upperPad.TotalSize() > 0))
RuntimeError("Padding isn't allowed for Unknown shape!");
if (op == PrimitiveOpType::Pooling)
kernelShape = operandShape.SubShape(0, inputRank-1);
else