diff options
Diffstat (limited to 'runtimes/pure_arm_compute/src/internal/MatrixSink.h')
| -rw-r--r-- | runtimes/pure_arm_compute/src/internal/MatrixSink.h | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/runtimes/pure_arm_compute/src/internal/MatrixSink.h b/runtimes/pure_arm_compute/src/internal/MatrixSink.h index 32bd49dc6..23ecc112b 100644 --- a/runtimes/pure_arm_compute/src/internal/MatrixSink.h +++ b/runtimes/pure_arm_compute/src/internal/MatrixSink.h @@ -14,6 +14,11 @@ * limitations under the License. */ +/** + * @file MatrixSink.h + * @ingroup COM_AI_RUNTIME + * @brief This file defines MatrixSink class + */ #ifndef __INTERNAL_MATRIX_SINK_H__ #define __INTERNAL_MATRIX_SINK_H__ @@ -27,9 +32,19 @@ #include <cstring> #include <cassert> +/** + * @brief Class to get matrix data from arm compute tensor + */ template <typename T> class MatrixSink final : public Sink { public: + /** + * @brief Construct a new Matrix Sink object + * @param[in] H Height of matrix + * @param[in] W Width of matrix + * @param[in] base Pointer to get data + * @param[in] size Size of matrix + */ MatrixSink(const int32_t H, const int32_t W, T *base, const size_t size) : _height{H}, _width{W}, _base{base} { @@ -37,6 +52,11 @@ public: } public: + /** + * @brief Get matrix data from arm compute tensor to base + * @param[in] tensor Tensor object of arm compute to get data + * @return N/A + */ void pull(::arm_compute::ITensor &tensor) const override { assert(tensor.info()->dimension(0) == _width); |