Method attributeAssignmentSpecialHandlingHack

source/python.js:13938–13969

(qualified_classname, assign)

            parseImports(/* p */) {
            }
            attributeAssignmentSpecialHandlingHack(qualified_classname, assign) {
                const replacements = new Map([
                    ['__torch__.torch.ao.nn.quantized.modules.linear.LinearPackedParams', ['_packed_params', 'Tensor', '__torch__.torch.classes.quantized.LinearPackedParamsBase']],
                    ['__torch__.torch.ao.nn.quantized.modules.linear.Linear', ['_packed_params', 'Tensor', '__torch__.torch.classes.quantized.LinearPackedParamsBase']],
                    ['__torch__.torch.ao.nn.quantized.dynamic.modules.linear.Linear', ['_packed_params', 'Tensor', '__torch__.torch.classes.quantized.LinearPackedParamsBase']],
                    ['__torch__.torch.ao.nn.quantized.modules.conv.Conv2d', ['_packed_params', 'Tensor', '__torch__.torch.classes.quantized.Conv2dPackedParamsBase']],
                    ['__torch__.torch.nn.intrinsic.quantized.modules.conv_relu.ConvReLU2d', ['_packed_params', 'Tensor', '__torch__.torch.classes.quantized.Conv2dPackedParamsBase']],
                    ['__torch__.torch.ao.nn.quantized.modules.conv.Conv3d', ['_packed_params', 'Tensor', '__torch__.torch.classes.quantized.Conv3dPackedParamsBase']],
                    ['__torch__.torch.nn.intrinsic.quantized.modules.conv_relu.ConvReLU3d', ['_packed_params', 'Tensor', '__torch__.torch.classes.quantized.Conv3dPackedParamsBase']],
                    ['__torch__.torch.nn.quantized.modules.linear.LinearPackedParams', ["_packed_params", "Tensor", "__torch__.torch.classes.quantized.LinearPackedParamsBase"]],
                    ["__torch__.torch.nn.quantized.modules.linear.Linear", ["_packed_params", "Tensor", "__torch__.torch.classes.quantized.LinearPackedParamsBase"]],
                    ["__torch__.torch.nn.quantized.modules.conv.Conv2d", ["_packed_params", "Tensor", "__torch__.torch.classes.quantized.Conv2dPackedParamsBase"]],
                    ["__torch__.torch.nn.quantized.modules.conv.Conv3d", ["_packed_params", "Tensor", "__torch__.torch.classes.quantized.Conv3dPackedParamsBase"]],
                    ["__torch__.torch.nn.quantized.dynamic.modules.linear.Linear", ["_packed_params", "Tensor", "__torch__.torch.classes.quantized.LinearPackedParamsBase"]]
                ]);
                const mangleRe = /\.___torch_mangle_\d+/g;
                const demangled_classname = qualified_classname.qualifiedName().replace(mangleRe, '');
                if (replacements.has(demangled_classname)) {
                    const lhs = assign.target;
                    if (!assign.annotation || assign.annotation instanceof ast.Name === false) {
                        return null;
                    }
                    const type = assign.annotation.id;
                    const [attr_name, expected_type, replacement_type] = replacements.get(demangled_classname);
                    if (lhs.id === attr_name && type === expected_type) {
                        const p = new torch._C.Parser(new torch._C.Source(replacement_type));
                        const typename_expr = p.parseExp();
                        return new ast.AnnAssign(lhs, typename_expr, assign.value, lhs instanceof ast.Name);
                    }
                }
                return null;
            }
            LEGACY_import_methods(mod, src) {
                const self = new torch._C.SimpleSelf(mod.type());
                const prefix = mod.type().name();