aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/graph/OperationRegistrar.h
diff options
context:
space:
mode:
authorGeorgios Pinitas <georgios.pinitas@arm.com>2017-10-25 18:26:46 +0100
committerAnthony Barbier <anthony.barbier@arm.com>2018-11-02 16:35:24 +0000
commit407c3e6e383affa7ebe72ce5f50fcf163ff037a3 (patch)
treec516e583fe0a2e5ee9dfc218fe10880000edcf91 /arm_compute/graph/OperationRegistrar.h
parente500747b5c1d27aeffae316c8190f6d169bb2fbd (diff)
downloadComputeLibrary-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/OperationRegistrar.h')
-rw-r--r--arm_compute/graph/OperationRegistrar.h8
1 files changed, 4 insertions, 4 deletions
diff --git a/arm_compute/graph/OperationRegistrar.h b/arm_compute/graph/OperationRegistrar.h
index ff31963dc3..ee171c3510 100644
--- a/arm_compute/graph/OperationRegistrar.h
+++ b/arm_compute/graph/OperationRegistrar.h
@@ -43,15 +43,15 @@ class OperationRegistrar final
public:
/** Add a new test case with the given name to the framework.
*
- * @param[in] operation_name Operation name
+ * @param[in] operation Operation type
*/
- OperationRegistrar(std::string operation_name);
+ OperationRegistrar(OperationType operation);
};
template <typename T>
-inline OperationRegistrar<T>::OperationRegistrar(std::string operation_name)
+inline OperationRegistrar<T>::OperationRegistrar(OperationType operation)
{
- OperationRegistry::get().add_operation<T>(std::move(operation_name));
+ OperationRegistry::get().add_operation<T>(std::move(operation));
}
} // namespace detail
} // namespace graph