diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2017-11-10 18:14:06 +0000 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:35:24 +0000 |
commit | 283c1790da45ab562ecfb2aa7741297191886d85 (patch) | |
tree | 45956bb79167e17aa634fd5f4d05c68ba059274c /src/core/CL/kernels/CLSoftmaxLayerKernel.cpp | |
parent | 624b77859dc9d0618056dad66833b9c37033337b (diff) | |
download | ComputeLibrary-283c1790da45ab562ecfb2aa7741297191886d85.tar.gz |
COMPMID-676: Rework TensorInfo building
Change-Id: Ic98f64ffe30739437a1fe31ef98d83ee900741e3
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/95512
Reviewed-by: Michalis Spyrou <michalis.spyrou@arm.com>
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'src/core/CL/kernels/CLSoftmaxLayerKernel.cpp')
-rw-r--r-- | src/core/CL/kernels/CLSoftmaxLayerKernel.cpp | 6 |
1 files changed, 1 insertions, 5 deletions
diff --git a/src/core/CL/kernels/CLSoftmaxLayerKernel.cpp b/src/core/CL/kernels/CLSoftmaxLayerKernel.cpp index af4fd88593..5331f40838 100644 --- a/src/core/CL/kernels/CLSoftmaxLayerKernel.cpp +++ b/src/core/CL/kernels/CLSoftmaxLayerKernel.cpp @@ -386,11 +386,7 @@ void CLLogits1DNormKernel::configure(const ICLTensor *input, const ICLTensor *su // Output auto initialization if not yet initialized auto_init_if_empty(*output->info(), - input->info()->tensor_shape(), - 1, - output_data_type, - input->info()->fixed_point_position(), - allowed_quantization_info); + input->info()->clone()->set_data_type(output_data_type).set_quantization_info(allowed_quantization_info)); ARM_COMPUTE_ERROR_ON_MISMATCHING_SHAPES(input, output); if(!is_quantized_asymmetric) |