diff options
Diffstat (limited to 'delegate/opaque/src/SharedFunctions.hpp')
| -rw-r--r-- | delegate/opaque/src/SharedFunctions.hpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/delegate/opaque/src/SharedFunctions.hpp b/delegate/opaque/src/SharedFunctions.hpp index 72fbe6a332..4698a1e989 100644 --- a/delegate/opaque/src/SharedFunctions.hpp +++ b/delegate/opaque/src/SharedFunctions.hpp @@ -23,5 +23,11 @@ TfLiteStatus ValidateFusedActivationOperator(DelegateData& delegateData, const armnn::TensorInfo& outputInfo, TfLiteFusedActivation activationType); +TfLiteOpaqueNode* GetNodeConnectedToInput(TfLiteOpaqueContext* tfLiteContext, + int32_t& connectedIndex, + int32_t inputIdx); + +bool WillInputBeOptimizedToConst(TfLiteOpaqueContext* tfLiteContext, int32_t inputIdx); + } // namespace armnnOpaqueDelegate |