diff options
Diffstat (limited to 'src/backends/reference/workloads/RefRankWorkload.hpp')
-rw-r--r-- | src/backends/reference/workloads/RefRankWorkload.hpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/backends/reference/workloads/RefRankWorkload.hpp b/src/backends/reference/workloads/RefRankWorkload.hpp index 237ae999ce..288dddd21d 100644 --- a/src/backends/reference/workloads/RefRankWorkload.hpp +++ b/src/backends/reference/workloads/RefRankWorkload.hpp @@ -32,7 +32,7 @@ private: { const int32_t rank = static_cast<int32_t>(GetTensorInfo(inputs[0]).GetNumDimensions()); - std::memcpy(GetOutputTensorData<void>(0, m_Data), &rank, sizeof(int32_t)); + std::memcpy(outputs[0]->Map(), &rank, sizeof(int32_t)); outputs[0]->Unmap(); } }; |