Method mark

source/python.js:6836–6882

(...args)

                return anyMarked;
            }
            mark(...args) {
                if (args.length === 1 && args[0] instanceof torch.Block) {
                    const [block] = args;
                    let anyMarked = false;
                    for (const node of block.nodes()) {
                        if (this._sideEffectPolicy === 'DONT_DELETE_NODES_WITH_SIDE_EFFECTS' && this.hasSideEffects(node)) {
                            const marked = this.mark(node);
                            anyMarked = anyMarked || marked;
                        }
                    }
                    const marked = this.markReturnNode(block.return_node());
                    anyMarked = anyMarked || marked;
                    for (const node of block.nodes()) {
                        if (node.kind() === 'prim::Loop') {
                            const marked = this.markLoop(node);
                            anyMarked = anyMarked || marked;
                        } else {
                            for (const subBlock of node.blocks()) {
                                const marked = this.mark(subBlock);
                                anyMarked = anyMarked || marked;
                            }
                        }
                        const marked = this.markIfLive(node);
                        anyMarked = anyMarked || marked;
                    }
                    return anyMarked;
                }
                if (args.length === 1 && args[0] instanceof torch.Node) {
                    const [node] = args;
                    if (this._marked.has(node)) {
                        return false;
                    }
                    this._marked.add(node);
                    let curNode = node;
                    while (curNode && curNode.owningBlock()) {
                        this.mark(curNode);
                        curNode = curNode.owningBlock().owningNode();
                    }
                    for (const input of node.inputs()) {
                        if (!this._liveValues.has(input)) {
                            this._liveValues.add(input);
                        }
                    }
                    return true;
                }
                throw new python.Error('Not implemented.');
            }
            markIfLive(node) {
                for (const output of node.outputs()) {
                    if (this._liveValues.has(output)) {