aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/graph/IGraphMutator.h
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/graph/IGraphMutator.h')
-rw-r--r--arm_compute/graph/IGraphMutator.h13
1 files changed, 13 insertions, 0 deletions
diff --git a/arm_compute/graph/IGraphMutator.h b/arm_compute/graph/IGraphMutator.h
index b80d37d274..94a28f63b1 100644
--- a/arm_compute/graph/IGraphMutator.h
+++ b/arm_compute/graph/IGraphMutator.h
@@ -35,6 +35,14 @@ class Graph;
class IGraphMutator
{
public:
+ /** Mutation type */
+ enum class MutationType
+ {
+ IR, /** IR specific mutation */
+ Backend /** Backend specific mutation */
+ };
+
+public:
/** Virtual Destructor */
virtual ~IGraphMutator() = default;
/** Walk the graph and perform a specific mutation
@@ -42,6 +50,11 @@ public:
* @param[in, out] g Graph to walk and mutate
*/
virtual void mutate(Graph &g) = 0;
+ /** Returns mutation type
+ *
+ * @return Mutation type enumeration
+ */
+ virtual MutationType type() const = 0;
/** Returns mutator name
*
* @return Mutator name