diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/core/NEON/kernels/NEGaussian5x5Kernel.cpp | 4 | ||||
-rw-r--r-- | src/runtime/NEON/functions/NEGaussian5x5.cpp | 2 |
2 files changed, 3 insertions, 3 deletions
diff --git a/src/core/NEON/kernels/NEGaussian5x5Kernel.cpp b/src/core/NEON/kernels/NEGaussian5x5Kernel.cpp index f872cc2f0a..4e1880d968 100644 --- a/src/core/NEON/kernels/NEGaussian5x5Kernel.cpp +++ b/src/core/NEON/kernels/NEGaussian5x5Kernel.cpp @@ -116,8 +116,8 @@ BorderSize NEGaussian5x5VertKernel::border_size() const void NEGaussian5x5VertKernel::configure(const ITensor *input, ITensor *output, bool border_undefined) { - ARM_COMPUTE_ERROR_ON_FORMAT_NOT_IN(input, Format::S16); - ARM_COMPUTE_ERROR_ON_FORMAT_NOT_IN(output, Format::U8); + ARM_COMPUTE_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(input, 1, DataType::S16); + ARM_COMPUTE_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(output, 1, DataType::U8); _input = input; _output = output; diff --git a/src/runtime/NEON/functions/NEGaussian5x5.cpp b/src/runtime/NEON/functions/NEGaussian5x5.cpp index 5ccc765966..69639d0d43 100644 --- a/src/runtime/NEON/functions/NEGaussian5x5.cpp +++ b/src/runtime/NEON/functions/NEGaussian5x5.cpp @@ -40,7 +40,7 @@ NEGaussian5x5::NEGaussian5x5() void NEGaussian5x5::configure(ITensor *input, ITensor *output, BorderMode border_mode, uint8_t constant_border_value) { // Init temporary buffer - TensorInfo tensor_info(input->info()->tensor_shape(), Format::S16); + TensorInfo tensor_info(input->info()->tensor_shape(), 1, DataType::S16); _tmp.allocator()->init(tensor_info); // Create and configure kernels for the two passes |