From bf3e36f9705d7372a2837bff72af0872ce735f04 Mon Sep 17 00:00:00 2001 From: Matthew Bentham Date: Fri, 25 Nov 2022 13:18:40 +0000 Subject: Add profile data for explicit copy of input and outputs Signed-off-by: Matthew Bentham Change-Id: I54d4a017e5f2d0bbeb742b6a33121e9ca5327f47 --- src/armnn/LoadedNetwork.cpp | 2 ++ 1 file changed, 2 insertions(+) (limited to 'src') diff --git a/src/armnn/LoadedNetwork.cpp b/src/armnn/LoadedNetwork.cpp index 7b24fd77b8..b42874f29d 100644 --- a/src/armnn/LoadedNetwork.cpp +++ b/src/armnn/LoadedNetwork.cpp @@ -1328,6 +1328,7 @@ void LoadedNetwork::EnqueueInput(const ConstTensor& inputTensor, ITensorHandle* } else { + ARMNN_SCOPED_PROFILING_EVENT(Compute::Undefined, "CopyInput"); std::unique_ptr tensorHandle = std::make_unique(inputTensor.GetInfo(), inputTensor.GetMemoryArea()); @@ -1374,6 +1375,7 @@ void LoadedNetwork::ImportOutputTensor(const Tensor& outputTensor, ITensorHandle void CopyToOutputTensor(const Tensor& outputTensor, ITensorHandle* outputTensorHandle) { + ARMNN_SCOPED_PROFILING_EVENT(Compute::Undefined, "CopyOutput"); auto copyFunc = [](void* dst, const void* src, size_t size) { memcpy(dst, src, size); -- cgit v1.2.1