diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2017-10-25 18:26:46 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:35:24 +0000 |
commit | 407c3e6e383affa7ebe72ce5f50fcf163ff037a3 (patch) | |
tree | c516e583fe0a2e5ee9dfc218fe10880000edcf91 /arm_compute/graph/OperationRegistry.h | |
parent | e500747b5c1d27aeffae316c8190f6d169bb2fbd (diff) | |
download | ComputeLibrary-407c3e6e383affa7ebe72ce5f50fcf163ff037a3.tar.gz |
COMPMID-630: Rework nodes
Reworked node:
-BatchNormalization
-Floor
-FullyConncted
-L2Normalize
-Normalization
-Pooling
-Softmax
Change-Id: I4c71cfffb1f59aac3326ba8b1f831339c5244394
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/93134
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'arm_compute/graph/OperationRegistry.h')
-rw-r--r-- | arm_compute/graph/OperationRegistry.h | 20 |
1 files changed, 10 insertions, 10 deletions
diff --git a/arm_compute/graph/OperationRegistry.h b/arm_compute/graph/OperationRegistry.h index 905e8ee66f..ae68bf45a2 100644 --- a/arm_compute/graph/OperationRegistry.h +++ b/arm_compute/graph/OperationRegistry.h @@ -47,39 +47,39 @@ public: static OperationRegistry &get(); /** Finds an operation in the registry * - * @param[in] operation Name of the operation to find + * @param[in] operation Type of the operation to find * @param[in] target Target of the operation * * @return Pointer to the operation functor if found, else nullptr */ - IOperation *find_operation(const std::string &operation, TargetHint target); + IOperation *find_operation(OperationType operation, TargetHint target); /** Checks if an operation for a given target exists * - * @param[in] operation Operation name + * @param[in] operation Operation type * @param[in] target Execution target * * @return True if exists else false */ - bool contains(const std::string &operation, TargetHint target) const; - /** Registers and operation to the registry + bool contains(OperationType operation, TargetHint target) const; + /** Registers an operation to the registry * - * @param operation_name Name of the operation to register + * @param operation Operation to register */ template <typename T> - void add_operation(const std::string &operation_name); + void add_operation(OperationType operation); private: /** Default Constructor */ OperationRegistry(); private: - std::map<std::string, std::vector<std::unique_ptr<IOperation>>> _registered_ops; + std::map<OperationType, std::vector<std::unique_ptr<IOperation>>> _registered_ops; }; template <typename T> -inline void OperationRegistry::add_operation(const std::string &operation_name) +inline void OperationRegistry::add_operation(OperationType operation) { - _registered_ops[operation_name].emplace_back(support::cpp14::make_unique<T>()); + _registered_ops[operation].emplace_back(support::cpp14::make_unique<T>()); } } // namespace graph } // namespace arm_compute |