Class TensorND

dnn/include/megdnn/basic_types.h:462–497

* \brief A simple encapsulation class for n-dimensional tensor. */

 * \brief A simple encapsulation class for n-dimensional tensor.
 */
struct TensorND {
    TensorLayout layout;

    TensorND() : m_ref_ptr(RefPtr((void*)nullptr)) {}

    TensorND(void* raw_ptr_, const TensorLayout& layout_)
            : layout(layout_), m_ref_ptr(raw_ptr_) {}

    TensorND(const TensorLayout& layout_, const RefPtr& ref_ptr)
            : layout(layout_), m_ref_ptr(ref_ptr) {}

    MGE_WIN_DECLSPEC_FUC void reset_ptr(void* ptr, size_t offset = 0);

    void* raw_ptr() const { return m_ref_ptr.get_ptr(); }

    const RefPtr get_ref_ptr() const { return m_ref_ptr; }

    RefPtr& get_ref_ptr() { return m_ref_ptr; }

    //! get typed pointer; type check is performed
    template <typename T>
    T* ptr() const {
        layout.dtype.assert_is_ctype<T>();
        return static_cast<T*>(m_ref_ptr.get_ptr());
    }

    //! get typed pointer of compatible type
    template <typename T>
    T* compatible_ptr() const {
        layout.dtype.assert_is_compatible_ctype<T>();
        return reinterpret_cast<T*>(m_ref_ptr.get_ptr());
    }

private:
    RefPtr m_ref_ptr;
};

#if MEGDNN_CC_HOST
using TensorFormat = TensorLayout::Format;