[mir_onnx] Fix is_test attribute checking (#6831)
authorСергей Баранников/AI Tools Lab /SRR/Engineer/삼성전자 <s.barannikov@samsung.com>
Fri, 23 Aug 2019 11:23:22 +0000 (20:23 +0900)
committerAlexander Efimov/AI Tools Lab/./Samsung Electronics <a.efimov@samsung.com>
Fri, 23 Aug 2019 11:23:22 +0000 (14:23 +0300)
Reverse `is_test` attribute testing. Only `is_test` mode is supported.

Signed-off-by: Sergei Barannikov <s.barannikov@samsung.com>
compiler/mir-onnx-importer/Op/BatchNormalization.cpp

index c47b733..cce1da8 100644 (file)
@@ -57,8 +57,8 @@ void BatchNormalizationNodeConverter::convertV1(const onnx::NodeProto &onnx_node
 void BatchNormalizationNodeConverter::convertV6(const onnx::NodeProto &onnx_node,
                                                 ConverterContext *context) const
 {
-  const auto is_test = getAttributeValue<int64_t>(onnx_node, "is_test", 0);
-  if (is_test != 0)
+  const auto is_test = getAttributeValue<std::int64_t>(onnx_node, "is_test", 0);
+  if (is_test == 0)
     throw std::runtime_error("Not supported is_test attribute!");
 
   convertV7(onnx_node, context);