From 81770986ac19d75c8b2ffebc87f97aaa81ad714f Mon Sep 17 00:00:00 2001 From: Richard Burton Date: Wed, 29 Apr 2020 20:07:09 +0100 Subject: Changing print statement to logging Signed-off-by: Richard Burton Change-Id: I38a929ec6ba8c6df0a84122a3329d80f86b5c0d0 --- python/pyarmnn/src/pyarmnn/_tensor/workload_tensors.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'python/pyarmnn') 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 -- cgit v1.2.1