diff options
Diffstat (limited to 'include/armnn/ILayerSupport.hpp')
| -rw-r--r-- | include/armnn/ILayerSupport.hpp | 14 |
1 files changed, 9 insertions, 5 deletions
diff --git a/include/armnn/ILayerSupport.hpp b/include/armnn/ILayerSupport.hpp index 54f4a2883b..a2d3961d34 100644 --- a/include/armnn/ILayerSupport.hpp +++ b/include/armnn/ILayerSupport.hpp @@ -111,11 +111,15 @@ public: const TensorInfo& output, Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0; - virtual bool IsDetectionPostProcessSupported( - const TensorInfo& input0, - const TensorInfo& input1, - const DetectionPostProcessDescriptor& descriptor, - Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const = 0; + virtual bool IsDetectionPostProcessSupported(const TensorInfo& boxEncodings, + const TensorInfo& scores, + const TensorInfo& anchors, + const TensorInfo& detectionBoxes, + const TensorInfo& detectionClasses, + const TensorInfo& detectionScores, + const TensorInfo& numDetections, + const DetectionPostProcessDescriptor& descriptor, + Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const =0; virtual bool IsDilatedDepthwiseConvolutionSupported( const TensorInfo& input, |