diff options
Diffstat (limited to 'src/backends/reference/workloads/RefInstanceNormalizationWorkload.cpp')
-rw-r--r-- | src/backends/reference/workloads/RefInstanceNormalizationWorkload.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/backends/reference/workloads/RefInstanceNormalizationWorkload.cpp b/src/backends/reference/workloads/RefInstanceNormalizationWorkload.cpp index daee97ae3e..e642dc9b9a 100644 --- a/src/backends/reference/workloads/RefInstanceNormalizationWorkload.cpp +++ b/src/backends/reference/workloads/RefInstanceNormalizationWorkload.cpp @@ -37,8 +37,9 @@ void RefInstanceNormalizationWorkload::Execute(std::vector<ITensorHandle*> input inputs[0]->Map()); std::unique_ptr<Encoder<float>> outputEncoder = MakeEncoder<float>(GetTensorInfo(outputs[0]), outputs[0]->Map()); + const TensorInfo& inputInfo = GetTensorInfo(inputs[0]); - InstanceNorm(m_Data, *inputDecoder, *outputEncoder); + InstanceNorm(m_Data, inputInfo, *inputDecoder, *outputEncoder); } } // namespace armnn |