aboutsummaryrefslogtreecommitdiff
path: root/src/backends/neon/workloads/NeonFloorFloatWorkload.cpp
diff options
context:
space:
mode:
authorKevin May <kevin.may@arm.com>2022-03-02 12:11:31 +0000
committerKevin May <kevin.may@arm.com>2022-03-02 12:11:31 +0000
commit998a20832fd511de75c2cd935b7a0e851e91156d (patch)
tree0839bcc53555b06887708cb56429735b8315472d /src/backends/neon/workloads/NeonFloorFloatWorkload.cpp
parent138a88cf568fbc420a310344903f7eb9dec8e1e3 (diff)
downloadarmnn-998a20832fd511de75c2cd935b7a0e851e91156d.tar.gz
Typo in Guide
* Only file changed is shim/BuildGuideShimSupportLibrary.md * All other files are merge squash from Arm NN 22.02 Signed-off-by: Kevin May <kevin.may@arm.com> Change-Id: Id82a6e9ac8abf74c1073c08744712f50e98dece0
Diffstat (limited to 'src/backends/neon/workloads/NeonFloorFloatWorkload.cpp')
-rw-r--r--src/backends/neon/workloads/NeonFloorFloatWorkload.cpp39
1 files changed, 39 insertions, 0 deletions
diff --git a/src/backends/neon/workloads/NeonFloorFloatWorkload.cpp b/src/backends/neon/workloads/NeonFloorFloatWorkload.cpp
index b97e3cef75..1d53245c5f 100644
--- a/src/backends/neon/workloads/NeonFloorFloatWorkload.cpp
+++ b/src/backends/neon/workloads/NeonFloorFloatWorkload.cpp
@@ -32,6 +32,45 @@ void NeonFloorFloatWorkload::Execute() const
ARMNN_SCOPED_PROFILING_EVENT_NEON_GUID("NeonFloorFloatWorkload_Execute", this->GetGuid());
m_Layer->run();
}
+
+void NeonFloorFloatWorkload::ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot)
+{
+ ITensorHandle* backupHandle = this->m_Data.m_Inputs[slot];
+ this->m_Data.m_Inputs[slot] = tensorHandle;
+ try
+ {
+ Reconfigure();
+ }
+ catch(armnn::UnimplementedException& e)
+ {
+ // Cannot reconfigure, revert the slot back and throw the exception.
+ this->m_Data.m_Inputs[slot] = backupHandle;
+ throw e;
+ }
+}
+
+// Replace output tensor handle with the given TensorHandle
+void NeonFloorFloatWorkload::ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot)
+{
+ ITensorHandle* backupHandle = this->m_Data.m_Inputs[slot];
+ this->m_Data.m_Inputs[slot] = tensorHandle;
+ try
+ {
+ Reconfigure();
+ }
+ catch(armnn::UnimplementedException& e)
+ {
+ // Cannot reconfigure, revert the slot back and throw the exception.
+ this->m_Data.m_Inputs[slot] = backupHandle;
+ throw e;
+ }
+}
+
+void NeonFloorFloatWorkload::Reconfigure()
+{
+ throw armnn::UnimplementedException("Reconfigure not implemented for this workload");
+}
+
} //namespace armnn