diff options
Diffstat (limited to 'compiler/circle-eval-diff/src/Tensor.h')
| -rw-r--r-- | compiler/circle-eval-diff/src/Tensor.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/compiler/circle-eval-diff/src/Tensor.h b/compiler/circle-eval-diff/src/Tensor.h index 65ab60638..d4f65d951 100644 --- a/compiler/circle-eval-diff/src/Tensor.h +++ b/compiler/circle-eval-diff/src/Tensor.h @@ -18,6 +18,7 @@ #define __CIRCLE_EVAL_DIFF_TENSOR_H__ #include <loco.h> +#include <luci/IR/CircleNodeDecl.h> #include <vector> @@ -76,6 +77,8 @@ private: std::vector<uint8_t> _data; }; +std::shared_ptr<Tensor> createEmptyTensor(const luci::CircleNode *node); + } // namespace circle_eval_diff #endif // __CIRCLE_EVAL_DIFF_TENSOR_H__ |