diff options
Diffstat (limited to 'compiler/moco-tf/src/Op/COpCall.h')
| -rw-r--r-- | compiler/moco-tf/src/Op/COpCall.h | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/compiler/moco-tf/src/Op/COpCall.h b/compiler/moco-tf/src/Op/COpCall.h index 0bb8a93c9..ea81d3a98 100644 --- a/compiler/moco-tf/src/Op/COpCall.h +++ b/compiler/moco-tf/src/Op/COpCall.h @@ -17,9 +17,12 @@ #ifndef __OP_COP_CALL_H__ #define __OP_COP_CALL_H__ +#include "GraphBuilder.h" +#include "GraphBuilderContext.h" + #include <moco/tf/Frontend.h> -#include <moco/Import/GraphBuilder.h> +#include <tensorflow/core/framework/graph.pb.h> namespace moco { |