21.05
|
Mutation pass to fuss nodes. More...
#include <NodeFusionMutator.h>
Public Member Functions | |
virtual void | mutate (Graph &g) override |
Walk the graph and perform a specific mutation. More... | |
MutationType | type () const override |
Returns mutation type. More... | |
const char * | name () override |
Returns mutator name. More... | |
Public Member Functions inherited from IGraphMutator | |
virtual | ~IGraphMutator ()=default |
Virtual Destructor. More... | |
Additional Inherited Members | |
Public Types inherited from IGraphMutator | |
enum | MutationType { IR, Backend } |
Mutation type. More... | |
Mutation pass to fuss nodes.
Definition at line 35 of file NodeFusionMutator.h.
|
overridevirtual |
Walk the graph and perform a specific mutation.
[in,out] | g | Graph to walk and mutate |
Implements IGraphMutator.
Definition at line 300 of file NodeFusionMutator.cpp.
References ActivationLayerInfo::ABS, ARM_COMPUTE_ERROR_ON, ActivationLayerInfo::BOUNDED_RELU, arm_compute::graph::CL, Tensor::desc(), Graph::edge(), ActivationLayerInfo::ELU, arm_compute::graph::detail::fuse_convolution_with_batch_normalization(), arm_compute::graph::detail::fuse_depthwise_convolution_with_batch_normalization(), ActivationLayerInfo::HARD_SWISH, ActivationLayerInfo::IDENTITY, ActivationLayerInfo::LEAKY_RELU, ActivationLayerInfo::LINEAR, ActivationLayerInfo::LOGISTIC, ActivationLayerInfo::LU_BOUNDED_RELU, INode::output(), Edge::producer(), arm_compute::QASYMM8, TensorDescriptor::quant_info, ActivationLayerInfo::RELU, ActivationLayerInfo::SOFT_RELU, ActivationLayerInfo::SQRT, ActivationLayerInfo::SQUARE, and ActivationLayerInfo::TANH.
|
overridevirtual |
Returns mutator name.
Implements IGraphMutator.
Definition at line 290 of file NodeFusionMutator.cpp.
|
overridevirtual |
Returns mutation type.
Implements IGraphMutator.
Definition at line 295 of file NodeFusionMutator.cpp.
References IGraphMutator::Backend.