diff options
Diffstat (limited to 'delegate/classic/src/SharedFunctions.hpp')
-rw-r--r-- | delegate/classic/src/SharedFunctions.hpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/delegate/classic/src/SharedFunctions.hpp b/delegate/classic/src/SharedFunctions.hpp index b03a63ded9..78aa3fef0e 100644 --- a/delegate/classic/src/SharedFunctions.hpp +++ b/delegate/classic/src/SharedFunctions.hpp @@ -21,5 +21,11 @@ TfLiteStatus ValidateFusedActivationOperator(DelegateData& delegateData, const armnn::TensorInfo& outputInfo, TfLiteFusedActivation activationType); +TfLiteNode* GetNodeConnectedToInput(TfLiteContext* tfLiteContext, + int32_t& connectedIndex, + int32_t inputIdx); + +bool WillInputBeOptimizedToConst(TfLiteContext* tfLiteContext, int32_t inputIdx); + } // namespace armnnDelegate |