diff options
Diffstat (limited to 'src/backends/reference/workloads/RefFullyConnectedWorkload.cpp')
| -rw-r--r-- | src/backends/reference/workloads/RefFullyConnectedWorkload.cpp | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/src/backends/reference/workloads/RefFullyConnectedWorkload.cpp b/src/backends/reference/workloads/RefFullyConnectedWorkload.cpp index c7a3d901e7..ac82db90e5 100644 --- a/src/backends/reference/workloads/RefFullyConnectedWorkload.cpp +++ b/src/backends/reference/workloads/RefFullyConnectedWorkload.cpp @@ -34,11 +34,11 @@ void RefFullyConnectedWorkload::PostAllocationConfigure() const TensorInfo& inputInfo = GetTensorInfo(m_Data.m_Inputs[0]); BOOST_ASSERT(inputInfo.GetNumDimensions() > 1); m_InputShape = inputInfo.GetShape(); - m_InputDecoder = MakeDecoder<float>(inputInfo, m_Data.m_Inputs[0]->Map()); + m_InputDecoder = MakeDecoder<float>(inputInfo); const TensorInfo& outputInfo = GetTensorInfo(m_Data.m_Outputs[0]); m_OutputShape = outputInfo.GetShape(); - m_OutputEncoder = MakeEncoder<float>(outputInfo, m_Data.m_Outputs[0]->Map()); + m_OutputEncoder = MakeEncoder<float>(outputInfo); m_NumActivations = 1; // Total number of activations in the input. for (unsigned int i = 1; i < inputInfo.GetNumDimensions(); i++) @@ -51,6 +51,9 @@ void RefFullyConnectedWorkload::Execute() const { ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, "RefFullyConnectedWorkload_Execute"); + m_InputDecoder->Reset(m_Data.m_Inputs[0]->Map()); + m_OutputEncoder->Reset(m_Data.m_Outputs[0]->Map()); + FullyConnected(m_InputShape, *m_InputDecoder, m_OutputShape, |