Commit 497e5d39 authored by Artur Wojcik's avatar Artur Wojcik Committed by Michał Karzyński

[ONNX] Fix compilation of ONNX library (#1494)

parent b11b9c69
......@@ -43,6 +43,7 @@ if (${CMAKE_VERSION} VERSION_LESS 3.2)
-DProtobuf_PROTOC_EXECUTABLE=${Protobuf_PROTOC_EXECUTABLE}
-DProtobuf_INCLUDE_DIR=${Protobuf_INCLUDE_DIR}
-DPROTOBUF_LIBRARY=${Protobuf_LIBRARY}
-DPROTOBUF_INCLUDE_DIR=${Protobuf_INCLUDE_DIR}
-DPROTOBUF_SRC_ROOT_FOLDER=${Protobuf_SRC_ROOT_FOLDER}
TMP_DIR "${EXTERNAL_PROJECTS_ROOT}/onnx/tmp"
STAMP_DIR "${EXTERNAL_PROJECTS_ROOT}/onnx/stamp"
......@@ -65,6 +66,7 @@ else()
-DProtobuf_PROTOC_EXECUTABLE=${Protobuf_PROTOC_EXECUTABLE}
-DProtobuf_INCLUDE_DIR=${Protobuf_INCLUDE_DIR}
-DPROTOBUF_LIBRARY=${Protobuf_LIBRARY}
-DPROTOBUF_INCLUDE_DIR=${Protobuf_INCLUDE_DIR}
-DPROTOBUF_SRC_ROOT_FOLDER=${Protobuf_SRC_ROOT_FOLDER}
TMP_DIR "${EXTERNAL_PROJECTS_ROOT}/onnx/tmp"
STAMP_DIR "${EXTERNAL_PROJECTS_ROOT}/onnx/stamp"
......@@ -88,6 +90,7 @@ else()
UPDATE_COMMAND ""
CMAKE_ARGS -DONNX_GEN_PB_TYPE_STUBS=OFF
-DProtobuf_PROTOC_EXECUTABLE=${Protobuf_PROTOC_EXECUTABLE}
-DProtobuf_LIBRARY=${Protobuf_LIBRARY}
-DProtobuf_INCLUDE_DIR=${Protobuf_INCLUDE_DIR}
TMP_DIR "${EXTERNAL_PROJECTS_ROOT}/onnx/tmp"
STAMP_DIR "${EXTERNAL_PROJECTS_ROOT}/onnx/stamp"
......
......@@ -39,17 +39,17 @@ namespace ngraph
auto input_b = inputs.at(1);
auto input_c = inputs.at(2);
double alpha{node.get_attribute_value<double>("alpha", 1)};
double beta{node.get_attribute_value<double>("beta", 1)};
double alpha = node.get_attribute_value<double>("alpha", 1);
double beta = node.get_attribute_value<double>("beta", 1);
auto trans_a{node.get_attribute_value<int64_t>("transA", 0)};
auto trans_b{node.get_attribute_value<int64_t>("transB", 0)};
auto trans_a = node.get_attribute_value<int64_t>("transA", 0);
auto trans_b = node.get_attribute_value<int64_t>("transB", 0);
if (trans_a)
if (trans_a != 0)
{
input_a = transpose(input_a);
}
if (trans_b)
if (trans_b != 0)
{
input_b = transpose(input_b);
}
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment