ITensorHandle.hpp

Go to the documentation of this file.

//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//
#pragma once

#include <armnn/MemorySources.hpp>

#include <boost/core/ignore_unused.hpp>

namespace armnn
{

class TensorShape;

class ITensorHandle
{
public:
    virtual ~ITensorHandle(){}

    virtual void Manage() = 0;

    virtual void Allocate() = 0;

    virtual ITensorHandle* GetParent() const = 0;

    virtual const void* Map(bool blocking=true) const = 0;

    virtual void Unmap() const = 0;

    void* Map(bool blocking=true)
    {
        return const_cast<void*>(static_cast<const ITensorHandle*>(this)->Map(blocking));
    }

    void Unmap()
    {
        return static_cast<const ITensorHandle*>(this)->Unmap();
    }

    virtual TensorShape GetStrides() const = 0;

    virtual TensorShape GetShape() const = 0;

    // Testing support to be able to verify and set tensor data content
    virtual void CopyOutTo(void* memory) const = 0;
    virtual void CopyInFrom(const void* memory) = 0;

    virtual unsigned int GetImportFlags() const { return 0; }

    virtual bool Import(void* memory, MemorySource source)
    {
        boost::ignore_unused(memory, source);
        return false;
    };
};

}