21.02
|
#include <GraphContext.h>
Public Member Functions | |
GraphContext () | |
Constructor. More... | |
~GraphContext () | |
Destructor. More... | |
GraphContext (const GraphContext &)=delete | |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
GraphContext (GraphContext &&)=default | |
Default move constructor. More... | |
GraphContext & | operator= (const GraphContext &)=delete |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
GraphContext & | operator= (GraphContext &&)=default |
Default move assignment operator. More... | |
const GraphConfig & | config () const |
Graph configuration accessor. More... | |
void | set_config (const GraphConfig &config) |
Sets graph configuration. More... | |
bool | insert_memory_management_ctx (MemoryManagerContext &&memory_ctx) |
Inserts a memory manager context. More... | |
MemoryManagerContext * | memory_management_ctx (Target target) |
Gets a memory manager context for a given target. More... | |
std::map< Target, MemoryManagerContext > & | memory_managers () |
Gets the memory managers map. More... | |
bool | insert_weights_management_ctx (WeightsManagerContext &&weights_ctx) |
Inserts a weights manager context. More... | |
WeightsManagerContext * | weights_management_ctx (Target target) |
Gets a weights manager context for a given target. More... | |
std::map< Target, WeightsManagerContext > & | weights_managers () |
Gets the weights managers map. More... | |
void | finalize () |
Finalizes memory managers in graph context. More... | |
Graph context.
Definition at line 57 of file GraphContext.h.
GraphContext | ( | ) |
Constructor.
Definition at line 34 of file GraphContext.cpp.
~GraphContext | ( | ) |
Destructor.
Definition at line 39 of file GraphContext.cpp.
References arm_compute::graph::release_default_graph_context().
|
delete |
Prevent instances of this class from being copied (As this class contains pointers)
|
default |
Default move constructor.
const GraphConfig & config | ( | ) | const |
Graph configuration accessor.
Definition at line 46 of file GraphContext.cpp.
Referenced by GraphManager::finalize_graph(), arm_compute::graph::backends::get_memory_manager(), arm_compute::graph::backends::get_weights_manager(), GraphContext::set_config(), NEDeviceBackend::setup_backend_context(), and CLDeviceBackend::setup_backend_context().
void finalize | ( | ) |
Finalizes memory managers in graph context.
Definition at line 102 of file GraphContext.cpp.
References ARM_COMPUTE_ERROR_ON.
Referenced by GraphManager::finalize_graph().
bool insert_memory_management_ctx | ( | MemoryManagerContext && | memory_ctx | ) |
Inserts a memory manager context.
[in] | memory_ctx | Memory manage context |
Definition at line 56 of file GraphContext.cpp.
References arm_compute::mlgo::parser::end(), and arm_compute::graph::UNSPECIFIED.
Referenced by NEDeviceBackend::setup_backend_context(), GCDeviceBackend::setup_backend_context(), and CLDeviceBackend::setup_backend_context().
bool insert_weights_management_ctx | ( | WeightsManagerContext && | weights_ctx | ) |
Inserts a weights manager context.
[in] | weights_ctx | Weights manager context |
Definition at line 78 of file GraphContext.cpp.
References arm_compute::mlgo::parser::end(), and GraphContext::weights_managers().
Referenced by NEDeviceBackend::setup_backend_context(), and CLDeviceBackend::setup_backend_context().
MemoryManagerContext * memory_management_ctx | ( | Target | target | ) |
Gets a memory manager context for a given target.
[in] | target | To retrieve the management context |
Definition at line 68 of file GraphContext.cpp.
References arm_compute::mlgo::parser::end().
Referenced by arm_compute::graph::detail::configure_transition_manager(), arm_compute::graph::backends::get_memory_manager(), NEDeviceBackend::setup_backend_context(), GCDeviceBackend::setup_backend_context(), and CLDeviceBackend::setup_backend_context().
std::map< Target, MemoryManagerContext > & memory_managers | ( | ) |
Gets the memory managers map.
Definition at line 73 of file GraphContext.cpp.
Referenced by arm_compute::graph::detail::call_all_tasks().
|
delete |
Prevent instances of this class from being copied (As this class contains pointers)
|
default |
Default move assignment operator.
void set_config | ( | const GraphConfig & | config | ) |
Sets graph configuration.
[in] | config | Configuration to use |
Definition at line 51 of file GraphContext.cpp.
References GraphContext::config().
Referenced by Stream::finalize().
WeightsManagerContext * weights_management_ctx | ( | Target | target | ) |
Gets a weights manager context for a given target.
[in] | target | To retrieve the weights management context |
Definition at line 92 of file GraphContext.cpp.
References arm_compute::mlgo::parser::end().
Referenced by arm_compute::graph::backends::get_weights_manager(), NEDeviceBackend::setup_backend_context(), and CLDeviceBackend::setup_backend_context().
std::map< Target, WeightsManagerContext > & weights_managers | ( | ) |
Gets the weights managers map.
Definition at line 97 of file GraphContext.cpp.
Referenced by GraphContext::insert_weights_management_ctx().