Compile error fixed by including header properly #296

Open
engcang wants to merge 1 commits from engcang/master into master
3 changed files with 3 additions and 2 deletions
+1
View File
@@ -1,5 +1,6 @@
*~ *~
build/ build/
installed/
.vscode/ .vscode/
*.bin *.bin
*.pyc *.pyc
+1 -1
View File
@@ -1,6 +1,6 @@
#pragma once #pragma once
#include <iostream> #include <iostream>
#include "tkDNN/tkdnn.h" #include "tkdnn.h"
namespace tk { namespace dnn { namespace tk { namespace dnn {
+1 -1
View File
@@ -7,7 +7,7 @@
#include "Layer.h" #include "Layer.h"
#include "NvInfer.h" #include "NvInfer.h"
#include <memory> #include <memory>
#include <tkDNN/kernels.h> #include "kernels.h"
#include <pluginsRT/ActivationLeakyRT.h> #include <pluginsRT/ActivationLeakyRT.h>
#include <pluginsRT/ActivationLogisticRT.h> #include <pluginsRT/ActivationLogisticRT.h>
#include <pluginsRT/ActivationMishRT.h> #include <pluginsRT/ActivationMishRT.h>