diff options
Diffstat (limited to 'arm_compute/core')
| -rw-r--r-- | arm_compute/core/KernelDescriptors.h | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/arm_compute/core/KernelDescriptors.h b/arm_compute/core/KernelDescriptors.h index 4a64032b14..19ac254c04 100644 --- a/arm_compute/core/KernelDescriptors.h +++ b/arm_compute/core/KernelDescriptors.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2019-2022 Arm Limited. + * Copyright (c) 2019-2023 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -117,6 +117,8 @@ struct DirectConvComputeKernelInfo int32_t n0{ 1 }; /**< Number of columns to be processed by the kernel */ int32_t k0{ 1 }; /**< Number of partial accumulations to be processed in a single iteration by the kernel */ bool export_weights_to_cl_image{ false }; /**< Flag to export the weights to cl_image */ + bool export_output_to_cl_image{ false }; /**< Flag to export the output to cl_image */ + bool export_input_to_cl_image{ false }; /**< Flag to export the input to cl_image */ }; /** Descriptor used by the softmax kernels */ |