diff options
Diffstat (limited to 'runtimes/pure_arm_compute/src/internal/arm_compute/Cast.h')
| -rw-r--r-- | runtimes/pure_arm_compute/src/internal/arm_compute/Cast.h | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/runtimes/pure_arm_compute/src/internal/arm_compute/Cast.h b/runtimes/pure_arm_compute/src/internal/arm_compute/Cast.h index 42b547feb..211a6ac87 100644 --- a/runtimes/pure_arm_compute/src/internal/arm_compute/Cast.h +++ b/runtimes/pure_arm_compute/src/internal/arm_compute/Cast.h @@ -100,6 +100,18 @@ const float scale = 0.0f, const int32_t zeroPoint = 0); /** + * @brief Cast from internal tensor info to tensor info object of arm compute + * @param[in] shape Tensor shape + * @param[in] type Tensor type of arm compute + * @param[in] scale Scale of tensor quantization + * @param[in] zeroPoint Zeropoint of tensor quantization + * @return TensorInfo object of arm compute + */ +::arm_compute::TensorInfo asTensorInfo(const ::arm_compute::TensorShape &shape, + const ::arm_compute::DataType &type, const float scale, + const int32_t zeroPoint); + +/** * @brief Set value to arm compute tensor with casting * @param[in] value Value to set * @param[out] to Target tensor of arm compute |