diff options
author | Richard Burton <richard.burton@arm.com> | 2020-04-29 20:07:09 +0100 |
---|---|---|
committer | Richard <richard.burton@arm.com> | 2020-05-05 12:34:15 +0000 |
commit | 81770986ac19d75c8b2ffebc87f97aaa81ad714f (patch) | |
tree | 1b41d822f0ba62a25c442ca8ef98e2a2aaccdca6 /python/pyarmnn/src | |
parent | 884746d2677e50c47b4bc70a2829a777344f73c4 (diff) | |
download | armnn-81770986ac19d75c8b2ffebc87f97aaa81ad714f.tar.gz |
Changing print statement to logging
Signed-off-by: Richard Burton <richard.burton@arm.com>
Change-Id: I38a929ec6ba8c6df0a84122a3329d80f86b5c0d0
Diffstat (limited to 'python/pyarmnn/src')
-rw-r--r-- | python/pyarmnn/src/pyarmnn/_tensor/workload_tensors.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/python/pyarmnn/src/pyarmnn/_tensor/workload_tensors.py b/python/pyarmnn/src/pyarmnn/_tensor/workload_tensors.py index b33356c65a..22b876896d 100644 --- a/python/pyarmnn/src/pyarmnn/_tensor/workload_tensors.py +++ b/python/pyarmnn/src/pyarmnn/_tensor/workload_tensors.py @@ -5,6 +5,7 @@ This file contains functions relating to WorkloadTensors. WorkloadTensors are the inputTensors and outputTensors that are consumed by IRuntime.EnqueueWorkload. """ from typing import Union, List, Tuple +import logging import numpy as np @@ -121,6 +122,6 @@ def workload_tensors_to_ndarray(workload_tensors: List[Tuple[int, Union[Tensor, arrays = [] for index, (_, tensor) in enumerate(workload_tensors): arrays.append(tensor.get_memory_area().reshape(list(tensor.GetShape()))) - print("Workload tensor {} shape: {}".format(index, tensor.GetShape())) + logging.info("Workload tensor {} shape: {}".format(index, tensor.GetShape())) return arrays |