Method load

imperative/python/megengine/utils/network_node.py:396–401

(cls, opr)

    @classmethod
    def load(cls, opr):
        obj = super(MatrixMul, cls).load(opr)
        dim1, dim2 = len(opr.inputs[0].shape), len(opr.inputs[1].shape)
        obj.params["dimA"] = dim1
        obj.params["dimB"] = dim2
        return obj


@register_flops(MatrixMul)