21 template<armnn::DataType dataType,
typename T = armnn::ResolveType<dataType>>
26 const std::array<unsigned int, 4> shapeData = { { 1u, 1u, 6u, 5u } };
29 std::vector<T> inputData =
40 ret.m_ExpectedData = inputData;
42 std::vector<T> actualOutput(tensorInfo.GetNumElements());
50 outputTensorHandle->Allocate();
55 const unsigned int origin[4] = {};
60 : std::move(inputTensorHandle);
63 : std::move(outputTensorHandle);
66 AddInputToWorkload(memCopyQueueDesc, workloadInfo, tensorInfo, workloadInput.get());
67 AddOutputToWorkload(memCopyQueueDesc, workloadInfo, tensorInfo, workloadOutput.get());
69 dstWorkloadFactory.
CreateMemCopy(memCopyQueueDesc, workloadInfo)->Execute();
72 ret.m_ActualData = actualOutput;
77 template<
typename SrcWorkloadFactory,
78 typename DstWorkloadFactory,
84 WorkloadFactoryHelper<SrcWorkloadFactory>::GetMemoryManager();
87 WorkloadFactoryHelper<DstWorkloadFactory>::GetMemoryManager();
89 SrcWorkloadFactory srcWorkloadFactory = WorkloadFactoryHelper<SrcWorkloadFactory>::GetFactory(srcMemoryManager);
90 DstWorkloadFactory dstWorkloadFactory = WorkloadFactoryHelper<DstWorkloadFactory>::GetFactory(dstMemoryManager);
92 return MemCopyTest<dataType>(srcWorkloadFactory, dstWorkloadFactory, withSubtensors);
virtual std::unique_ptr< IWorkload > CreateMemCopy(const MemCopyQueueDescriptor &descriptor, const WorkloadInfo &info) const
#define ARMNN_NO_DEPRECATE_WARN_BEGIN
virtual std::unique_ptr< ITensorHandle > CreateSubTensorHandle(ITensorHandle &parent, TensorShape const &subTensorShape, unsigned int const *subTensorOrigin) const =0
typename ResolveTypeImpl< DT >::Type ResolveType
#define ARMNN_NO_DEPRECATE_WARN_END
std::shared_ptr< IMemoryManager > IMemoryManagerSharedPtr
void AllocateAndCopyDataToITensorHandle(armnn::ITensorHandle *tensorHandle, const void *memory)
void CopyDataFromITensorHandle(void *memory, const armnn::ITensorHandle *tensorHandle)
virtual std::unique_ptr< ITensorHandle > CreateTensorHandle(const TensorInfo &tensorInfo, const bool IsMemoryManaged=true) const =0
Contains information about TensorInfos of a layer.
virtual bool SupportsSubTensors() const =0