Function propagateModdimsShape

src/backend/cpu/kernel/Array.hpp:47–66

Sets the shape of the buffer node_index_map under the moddims node to the new shape

/// Sets the shape of the buffer node_index_map under the moddims node to the
/// new shape
void propagateModdimsShape(
    std::vector<std::shared_ptr<common::Node>> &node_clones) {
    using arrayfire::common::NodeIterator;
    for (auto &node : node_clones) {
        if (node->getOp() == af_moddims_t) {
            common::ModdimNode *mn =
                static_cast<common::ModdimNode *>(node.get());

            NodeIterator<> it(node.get());
            while (it != NodeIterator<>()) {
                it = std::find_if(it, NodeIterator<>(), common::isBuffer);
                if (it == NodeIterator<>()) { break; }

                it->setShape(mn->m_new_shape);

                ++it;
            }
        }
    }
}

/// Removes node_index_map whos operation matchs a unary operation \p op.
void removeNodeOfOperation(