blob: a6dc7a01d37a9803285a722e657772f7cd10d6dc (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
|
function(_ONNXProto_import)
nncc_find_package(ONNXSource EXACT 1.3.0 QUIET)
if(NOT ONNXSource_FOUND)
set(ONNXProto_FOUND FALSE PARENT_SCOPE)
return()
endif(NOT ONNXSource_FOUND)
nncc_find_package(Protobuf QUIET)
if(NOT Protobuf_FOUND)
set(ONNXProto_FOUND FALSE PARENT_SCOPE)
return()
endif(NOT Protobuf_FOUND)
if(NOT TARGET onnxproto)
nncc_include(ExternalProjectTools)
add_extdirectory("${CMAKE_CURRENT_LIST_DIR}/ONNXProto" onnxproto)
endif(NOT TARGET onnxproto)
message(STATUS "Found ONNX: TRUE")
set(ONNXProto_FOUND TRUE PARENT_SCOPE)
endfunction(_ONNXProto_import)
_ONNXProto_import()
|