From 4536193e78570989c1b54f1c5d57627f29f9d400 Mon Sep 17 00:00:00 2001 From: Michele Di Giorgio Date: Thu, 19 Dec 2019 13:53:44 +0000 Subject: COMPMID-2801: Add support for QASYMM8_SIGNED in NEDirectConvolutionLayerOutputStageKernel Change-Id: Ib047dd1024b8ecac60e2d368cb161ca418c933ff Signed-off-by: Michele Di Giorgio Reviewed-on: https://review.mlplatform.org/c/2503 Reviewed-by: Giorgio Arena Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins Reviewed-by: Georgios Pinitas --- arm_compute/core/KernelDescriptors.h | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) (limited to 'arm_compute/core/KernelDescriptors.h') diff --git a/arm_compute/core/KernelDescriptors.h b/arm_compute/core/KernelDescriptors.h index f358153b0d..d009ccc73d 100644 --- a/arm_compute/core/KernelDescriptors.h +++ b/arm_compute/core/KernelDescriptors.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2019 ARM Limited. + * Copyright (c) 2019-2020 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -83,5 +83,14 @@ struct SoftmaxKernelInfo bool is_log{ false }; /**< Flag used to perform Log Softmax operation */ DataType input_data_type{ DataType::UNKNOWN }; /**< Input tensor data type */ }; + +/** Descriptor used by the direct convolution layer output stage kernels */ +struct DirectConvolutionLayerOutputStageKernelInfo +{ + int32_t result_fixedpoint_multiplier{ 0 }; /**< Result output stage multiplier used for quantizing */ + int32_t result_shift{ 0 }; /**< Result output stage shift used for quantizing */ + int32_t result_offset_after_shift{ 0 }; /**< Result offset used for quantizing */ + DataType output_data_type{ DataType::UNKNOWN }; /**< Output tensor data type to use if the output is not initialized */ +}; } // namespace arm_compute #endif /* ARM_COMPUTE_CORE_KERNEL_DESCRIPTORS_H */ -- cgit v1.2.1