blob: 50f2642669d7100d467f614b0742dd094b100f01 [file] [log] [blame]
telsoa014fcda012018-03-09 14:13:49 +00001//
Colm Donelana98e79a2022-12-06 21:32:29 +00002// Copyright © 2017,2022 Arm Ltd and Contributors. All rights reserved.
David Beckecb56cd2018-09-05 12:52:57 +01003// SPDX-License-Identifier: MIT
telsoa014fcda012018-03-09 14:13:49 +00004//
5
Colm Donelana98e79a2022-12-06 21:32:29 +00006#include <armnn/backends/ITensorHandle.hpp>
Sadik Armagana097d2a2021-11-24 15:47:28 +00007#include <armnnTestUtils/TensorCopyUtils.hpp>
Aron Virginas-Tarc9cc8042018-11-01 16:15:57 +00008#include <Half.hpp>
arovir01616e7752018-10-01 17:08:59 +01009
David Beck09e2f272018-10-30 11:38:41 +000010void CopyDataToITensorHandle(armnn::ITensorHandle* tensorHandle, const void* memory)
telsoa014fcda012018-03-09 14:13:49 +000011{
David Beck09e2f272018-10-30 11:38:41 +000012 tensorHandle->CopyInFrom(memory);
telsoa014fcda012018-03-09 14:13:49 +000013}
14
David Beck09e2f272018-10-30 11:38:41 +000015void CopyDataFromITensorHandle(void* memory, const armnn::ITensorHandle* tensorHandle)
telsoa014fcda012018-03-09 14:13:49 +000016{
David Beck09e2f272018-10-30 11:38:41 +000017 tensorHandle->CopyOutTo(memory);
telsoa014fcda012018-03-09 14:13:49 +000018}
19
David Beck09e2f272018-10-30 11:38:41 +000020void AllocateAndCopyDataToITensorHandle(armnn::ITensorHandle* tensorHandle, const void* memory)
telsoa014fcda012018-03-09 14:13:49 +000021{
22 tensorHandle->Allocate();
David Beck09e2f272018-10-30 11:38:41 +000023 CopyDataToITensorHandle(tensorHandle, memory);
telsoa014fcda012018-03-09 14:13:49 +000024}