2020-03-20 22:40:51 +08:00
|
|
|
add_library(OMONNXToKrnl
|
|
|
|
ONNXToKrnlCommon.cpp
|
|
|
|
ONNXToKrnlCommon.hpp
|
|
|
|
Math/Elementwise.cpp
|
|
|
|
Math/Gemm.cpp
|
|
|
|
Math/MatMul.cpp
|
|
|
|
Math/Reduction.cpp
|
|
|
|
Math/Softmax.cpp
|
|
|
|
NN/Conv.cpp
|
|
|
|
NN/Normalization.cpp
|
|
|
|
NN/Pooling.cpp
|
2020-05-13 21:08:06 +08:00
|
|
|
RNN/RNNBase.cpp
|
|
|
|
RNN/RNNBase.hpp
|
|
|
|
RNN/LSTM.cpp
|
2020-03-20 22:40:51 +08:00
|
|
|
Tensor/Identity.cpp
|
|
|
|
Tensor/Reshape.cpp
|
|
|
|
Tensor/PadConstantValuePad.cpp
|
2020-05-15 13:19:28 +08:00
|
|
|
Tensor/Pad.cpp
|
2020-03-20 22:40:51 +08:00
|
|
|
Tensor/Transpose.cpp
|
2020-07-03 16:26:41 +08:00
|
|
|
Tensor/Squeeze.cpp
|
2020-03-20 22:40:51 +08:00
|
|
|
Tensor/Unsqueeze.cpp
|
|
|
|
Tensor/Constant.cpp
|
2020-04-13 23:40:39 +08:00
|
|
|
Tensor/Concat.cpp
|
2020-06-11 10:57:20 +08:00
|
|
|
Tensor/Split.cpp
|
2020-09-12 03:36:23 +08:00
|
|
|
Tensor/Gather.cpp
|
2020-03-20 22:40:51 +08:00
|
|
|
ConvertONNXToKrnl.cpp)
|
2020-06-04 21:05:04 +08:00
|
|
|
target_link_libraries(OMONNXToKrnl
|
|
|
|
onnx)
|
2020-03-20 22:40:51 +08:00
|
|
|
target_include_directories(OMONNXToKrnl
|
|
|
|
PRIVATE
|
|
|
|
${ONNX_MLIR_SRC_ROOT}
|
|
|
|
${ONNX_MLIR_BIN_ROOT}
|
|
|
|
${ONNX_MLIR_SRC_ROOT})
|
2020-04-14 17:40:05 +08:00
|
|
|
|
2020-06-09 10:38:32 +08:00
|
|
|
# Header dependencies
|
|
|
|
add_dependencies(OMONNXToKrnl OMKrnlOpsInc OMONNXOpsInc)
|
|
|
|
# Linking dependencies
|
|
|
|
add_dependencies(OMONNXToKrnl OMKrnlOps)
|