Method mixin_init_output_comp_node

src/core/impl/graph/operator_node.cpp:513–519

/* ===================== mixins::SingleCNOperatorNode =====================  */

void mixin::SingleCNOperatorNode::mixin_init_output_comp_node(OperatorNodeBase& opr) {
    auto cn = mixin_infer_output_comp_node(
            opr, opr.node_prop().contain(NodeProp::Flag::CROSS_COMP_NODE_MEMORY));
    for (auto i : opr.output())
        i->comp_node(cn);
    opr.on_output_comp_node_stream_changed();
}

CompNode mixin::SingleCNOperatorNode::mixin_infer_output_comp_node(
        const OperatorNodeBase& opr, bool cross_mem) {