9 #include <armnnTestUtils/MockMemoryManager.hpp> 16 return "Arm/Mock/TensorHandleFactory";
24 : m_MemoryManager(mgr)
31 unsigned int const* subTensorOrigin)
const override;
39 const bool IsMemoryManaged)
const override;
43 const bool IsMemoryManaged)
const override;
56 mutable std::shared_ptr<MockMemoryManager> m_MemoryManager;
static const FactoryId & GetIdStatic()
MemorySourceFlags GetImportFlags() const override
bool SupportsSubTensors() const override
unsigned int MemorySourceFlags
Copyright (c) 2021 ARM Limited and Contributors.
std::unique_ptr< ITensorHandle > CreateTensorHandle(const TensorInfo &tensorInfo) const override
constexpr const char * MockTensorHandleFactoryId()
MockTensorHandleFactory(std::shared_ptr< MockMemoryManager > mgr)
const FactoryId & GetId() const override
MemorySource
Define the Memory Source to reduce copies.
MemorySourceFlags GetExportFlags() const override
std::unique_ptr< ITensorHandle > CreateSubTensorHandle(ITensorHandle &parent, TensorShape const &subTensorShape, unsigned int const *subTensorOrigin) const override