diff options
Diffstat (limited to 'src/armnn/layers/DetectionPostProcessLayer.hpp')
-rw-r--r-- | src/armnn/layers/DetectionPostProcessLayer.hpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/armnn/layers/DetectionPostProcessLayer.hpp b/src/armnn/layers/DetectionPostProcessLayer.hpp index e40966a19c..b409134c1c 100644 --- a/src/armnn/layers/DetectionPostProcessLayer.hpp +++ b/src/armnn/layers/DetectionPostProcessLayer.hpp @@ -10,14 +10,14 @@ namespace armnn { -class ScopedCpuTensorHandle; +class ScopedTensorHandle; /// This layer represents a detection postprocess operator. class DetectionPostProcessLayer : public LayerWithParameters<DetectionPostProcessDescriptor> { public: /// A unique pointer to store Anchor values. - std::shared_ptr<ConstCpuTensorHandle> m_Anchors; + std::shared_ptr<ConstTensorHandle> m_Anchors; /// Makes a workload for the DetectionPostProcess type. /// @param [in] graph The graph where this layer can be found. |