Method mutate

src/graph/mutators/InPlaceOperationMutator.cpp:223–276

}

void InPlaceOperationMutator::mutate(Graph &g)
{
    std::set<NodeType> in_place_nodes = {NodeType::ActivationLayer,
                                         NodeType::BatchNormalizationLayer,
                                         NodeType::EltwiseLayer,
                                         NodeType::UnaryEltwiseLayer,
                                         NodeType::DepthwiseConvolutionLayer,
                                         NodeType::FusedDepthwiseConvolutionBatchNormalizationLayer,
                                         NodeType::PrintLayer};

    // Not interested in the order of nodes
    for (auto &node : g.nodes())
    {
        if (node && in_place_nodes.find(node->type()) != std::end(in_place_nodes))
        {
            // Get input edge
            Edge *input_edge = node->input_edge(0);

            // Check if parent has a single output if yes then force in place calculation else not
            if ((input_edge != nullptr) && output_edges_are_separate_tensors(g, input_edge))
            {
                if (node->type() == NodeType::EltwiseLayer)
                {
                    try_in_place_elementwise(node);
                }
                else if (node->type() == NodeType::FusedDepthwiseConvolutionBatchNormalizationLayer ||
                         node->type() == NodeType::DepthwiseConvolutionLayer)
                {
                    try_in_place_depthwiseconv(node);
                }
                else
                {
                    // Get current and new output tensors
                    auto current_output_tensor = node->output(0);
                    auto new_output_tensor     = input_edge->tensor();

                    ARM_COMPUTE_ERROR_ON(current_output_tensor == nullptr || new_output_tensor == nullptr);

                    // Prevent in-place operation if there is an accessor bound to the in-place tensor or quantization info are different
                    if (new_output_tensor->accessor() != nullptr ||
                        current_output_tensor->desc().quant_info != new_output_tensor->desc().quant_info)
                    {
                        ARM_COMPUTE_LOG_GRAPH_VERBOSE("Prevented in-place operation as there is an accessor bound to "
                                                      "the input tensor or the quantization info are different.\n");
                    }
                    else
                    {
                        set_new_output_and_inherit_accessor(node, current_output_tensor, new_output_tensor);
                    }
                }
            }
        }
    }
}
} // namespace graph
} // namespace arm_compute