diff options
Diffstat (limited to 'src/runtime/OperatorTensor.cpp')
-rw-r--r-- | src/runtime/OperatorTensor.cpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/runtime/OperatorTensor.cpp b/src/runtime/OperatorTensor.cpp index 5a35154479..a8ad53da90 100644 --- a/src/runtime/OperatorTensor.cpp +++ b/src/runtime/OperatorTensor.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2020 Arm Limited. + * Copyright (c) 2020-2021 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -24,6 +24,8 @@ #include "arm_compute/runtime/OperatorTensor.h" #include "arm_compute/runtime/MemoryRegion.h" +#include "support/Cast.h" + namespace arm_compute { namespace experimental @@ -48,7 +50,7 @@ uint8_t *OperatorTensor::buffer() const switch(_mem_type) { case MemoryType::CPU: - return (uint8_t *)dynamic_cast<MemoryRegion *>(_memory->region())->buffer(); + return (uint8_t *)utils::cast::polymorphic_downcast<MemoryRegion *>(_memory->region())->buffer(); default: ARM_COMPUTE_ERROR("Memory type not supported."); } |