diff options
Diffstat (limited to 'src/backends/reference/workloads/ResizeBilinear.hpp')
| -rw-r--r-- | src/backends/reference/workloads/ResizeBilinear.hpp | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/src/backends/reference/workloads/ResizeBilinear.hpp b/src/backends/reference/workloads/ResizeBilinear.hpp index 847b8e8bef..92b229d3bb 100644 --- a/src/backends/reference/workloads/ResizeBilinear.hpp +++ b/src/backends/reference/workloads/ResizeBilinear.hpp @@ -10,6 +10,10 @@ namespace armnn { -void ResizeBilinear(const float* in, const TensorInfo& inputInfo, float* out, const TensorInfo& outputInfo); +void ResizeBilinear(const float* in, + const TensorInfo& inputInfo, + float* out, + const TensorInfo& outputInfo, + DataLayoutIndexed dataLayout = DataLayout::NCHW); } //namespace armnn |