Method assignExitContinuations

source/python.js:18311–18355

(block)

                this._curr_loop = null;
            }
            assignExitContinuations(block) {
                for (const n of block.nodes()) {
                    switch (n.kind()) {
                        case 'prim::If': {
                            this.assignExitContinuations(n.blocks().at(0));
                            this.assignExitContinuations(n.blocks().at(1));
                            break;
                        }
                        case 'prim::Closure': {
                            const closure_block = new torch._C.LoopContinuations();
                            closure_block.run(n.blocks().at(0));
                            break;
                        }
                        case 'prim::Loop': {
                            const prev_loop = this._curr_loop;
                            this._curr_loop = n;
                            this.assignExitContinuations(n.blocks().at(0));
                            this._curr_loop = prev_loop;
                            break;
                        }
                        case 'prim::ContinueStmt': {
                            const loop_continuation = this._graph.create('prim::LoopContinuation', 0).insertAfter(n);
                            const header_block = loop_continuation.addBlock();
                            const [, pre_header] = this._curr_loop.blocks();
                            header_block.cloneFrom(pre_header, (v) => v);
                            this.InlineBlockBeforeNode(n, header_block);
                            loop_continuation.addInput(header_block.outputs()[0]);
                            loop_continuation.eraseBlock(0);
                            this.addLoopCarriedOutputs(loop_continuation);
                            n.destroy();
                            break;
                        }
                        case 'prim::BreakStmt': {
                            const loop_exit = this._graph.create('prim::LoopContinuation', 0).insertAfter(n);
                            loop_exit.addInput(this._false_val);
                            this.addLoopCarriedOutputs(loop_exit);
                            n.destroy();
                            break;
                        }
                        default: {
                            break;
                        }
                    }
                }
            }
            run(...args) {
                if (args.length === 1 && args[0] instanceof torch.Graph) {
                    const [graph] = args;