From 1fa27ad2f3047a1e466f5757d64938ee8db90957 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Thu, 22 Jul 2021 01:25:57 +0100 Subject: Inject temporary tensors to pack in they don't exist in CpuSoftmax Signed-off-by: Georgios Pinitas Change-Id: Ief95ad8c97271f9042534e149fbccc9199318c2a Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5975 Tested-by: Arm Jenkins Reviewed-by: Michele Di Giorgio Comments-Addressed: Arm Jenkins --- src/runtime/cpu/operators/CpuSoftmax.cpp | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'src/runtime') diff --git a/src/runtime/cpu/operators/CpuSoftmax.cpp b/src/runtime/cpu/operators/CpuSoftmax.cpp index e17925ee50..abbc539b19 100644 --- a/src/runtime/cpu/operators/CpuSoftmax.cpp +++ b/src/runtime/cpu/operators/CpuSoftmax.cpp @@ -160,11 +160,11 @@ void CpuSoftmaxGeneric::run(ITensorPack &tensors) auto src = tensors.get_const_tensor(TensorType::ACL_SRC); auto dst = tensors.get_tensor(TensorType::ACL_DST); - CpuAuxTensorHandler tmp(offset_int_vec(InternalTensorIdx::TMP), _tmp, tensors, false); - CpuAuxTensorHandler max(offset_int_vec(InternalTensorIdx::MAX), _max, tensors, false); + CpuAuxTensorHandler tmp(offset_int_vec(InternalTensorIdx::TMP), _tmp, tensors, true); + CpuAuxTensorHandler max(offset_int_vec(InternalTensorIdx::MAX), _max, tensors, true); - CpuAuxTensorHandler input_permuted(offset_int_vec(InternalTensorIdx::PERMUTED_SRC), _input_permuted, tensors, false); - CpuAuxTensorHandler output_permuted(offset_int_vec(InternalTensorIdx::PERMUTED_DST), _output_permuted, tensors, false); + CpuAuxTensorHandler input_permuted(offset_int_vec(InternalTensorIdx::PERMUTED_SRC), _input_permuted, tensors, true); + CpuAuxTensorHandler output_permuted(offset_int_vec(InternalTensorIdx::PERMUTED_DST), _output_permuted, tensors, true); ITensorPack max_pack; ITensorPack softmax_pack; -- cgit v1.2.1