Method save_graph_module

source/python.js:19840–19865

()

                this.module = new torch.nn.Module();
            }
            save_graph_module() {
                const Context = class {
                    constructor(self) {
                        this.self = self;
                    }
                    __enter__() {
                        this.saved = [
                            this.self.graph,
                            this.self.module,
                            this.self.serialized_name_to_node,
                            this.self.serialized_name_to_meta,
                            this.self.unbacked_symbols,
                        ];
                        this.self.graph = new torch.fx.graph.Graph();
                        this.self.module = new torch.nn.modules.module.Module();
                        this.self.serialized_name_to_node = new builtins.dict();
                        this.self.serialized_name_to_meta = new builtins.dict(); // torch._export.serde.serialize.LazyMap
                        this.self.unbacked_symbols = new Set();
                    }
                    __exit__(/* exc_type, exc_value, traceback */) {
                        const self = this.self;
                        [self.graph, self.module, self.serialized_name_to_node, self.serialized_name_to_meta, self.unbacked_symbols] = this.saved;
                    }
                };
                return new Context(this);
            }
            deserialize_graph_output(output) {
                if (output.type === 'as_tensor') {
                    return this.serialized_name_to_node.__getitem__(output.as_tensor.name);