Check groups attribute existence in convolution op ONNX export.

This commit is contained in:
Spandan Tiwari 2018-06-06 16:08:04 -07:00 коммит произвёл Thiago Crepaldi
Родитель 2b326401c0
Коммит f2192a145c
1 изменённых файлов: 1 добавлений и 1 удалений

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

@ -3198,7 +3198,7 @@ void CNTKToONNXHelper::CopyAttributes(const FunctionPtr& src, ONNXIR::Node* node
auto autoPadding = AsVector<bool>(src->Attributes()[L"autoPadding"].Value<std::vector<DictionaryValue>>());
auto dilations = (NDShape)src->Attributes()[L"dilation"].Value<NDShape>();
auto transpose = (bool)src->Attributes()[L"transpose"].Value<bool>();
auto groups = (size_t)src->Attributes()[L"groups"].Value<size_t>();
size_t groups = (src->Attributes().Contains(L"groups")) ? (size_t)src->Attributes()[L"groups"].Value<size_t>() : 1u;
//
// Remove the channel part for ONNX. This is because ONNX, unlike CNTK, does