24.02.1
|
Mutation pass to implement/optimize grouped convolutions. More...
#include <GroupedConvolutionMutator.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 implement/optimize grouped convolutions.
Definition at line 37 of file GroupedConvolutionMutator.h.
|
overridevirtual |
Walk the graph and perform a specific mutation.
[in,out] | g | Graph to walk and mutate |
Implements IGraphMutator.
Definition at line 120 of file GroupedConvolutionMutator.cpp.
References Graph::add_connection(), ARM_COMPUTE_ERROR_ON, INode::assigned_target(), arm_compute::test::validation::conv_info, arm_compute::graph::ConvolutionLayer, arm_compute::graph::EmptyNodeID, arm_compute::utility::for_each(), BackendRegistry::get(), BackendRegistry::get_backend(), arm_compute::graph::get_driving_nodes(), INode::id(), Graph::node(), Graph::nodes(), arm_compute::test::validation::num_groups, INode::output(), Graph::remove_node(), Tensor::set_accessor(), tf_frozen_model_extractor::t, Graph::tensors(), INode::type(), and IDeviceBackend::validate_node().
|
overridevirtual |
Returns mutator name.
Implements IGraphMutator.
Definition at line 110 of file GroupedConvolutionMutator.cpp.
|
overridevirtual |
Returns mutation type.
Implements IGraphMutator.
Definition at line 115 of file GroupedConvolutionMutator.cpp.
References IGraphMutator::Backend.