Function removeNodeOfOperation

src/backend/cpu/kernel/Array.hpp:69–91

Removes node_index_map whos operation matchs a unary operation \p op.

/// Removes node_index_map whos operation matchs a unary operation \p op.
void removeNodeOfOperation(
    std::vector<std::shared_ptr<common::Node>> &node_index_map, af_op_t op) {
    using arrayfire::common::Node;

    for (size_t nid = 0; nid < node_index_map.size(); nid++) {
        auto &node = node_index_map[nid];

        for (int i = 0;
             i < Node::kMaxChildren && node->m_children[i] != nullptr; i++) {
            if (node->m_children[i]->getOp() == op) {
                // replace moddims
                auto moddim_node    = node->m_children[i];
                node->m_children[i] = moddim_node->m_children[0];
            }
        }
    }

    node_index_map.erase(remove_if(begin(node_index_map), end(node_index_map),
                                   [op](std::shared_ptr<Node> &node) {
                                       return node->getOp() == op;
                                   }),
                         end(node_index_map));
}

/// Returns the cloned output_nodes located in the node_clones array
///