Method addIfLoadStores

source/python.js:18228–18251

(n)

                n.addInput(inp);
            }
            addIfLoadStores(n) {
                const [true_block, false_block] = n.blocks();
                const true_vars = this.addControlFlowLoadStores(true_block);
                const false_vars = this.addControlFlowLoadStores(false_block);
                const mutated_variables = new Set();
                for (const v of true_vars.definedVariables()) {
                    if (false_vars.findInAnyFrame(v)) {
                        mutated_variables.add(v);
                    }
                }
                for (const v of false_vars.definedVariables()) {
                    if (true_vars.findInAnyFrame(v)) {
                        mutated_variables.add(v);
                    }
                }
                for (const x of mutated_variables) {
                    const true_type = true_vars.findInAnyFrame(x);
                    const false_type = false_vars.findInAnyFrame(x);
                    const unified = torch._C.unifyTypes(true_type, false_type, /*default_to_union=*/true);
                    this.addBlockOutput(true_block, true_type, x);
                    this.addBlockOutput(false_block, false_type, x);
                    this.addNodeOutput(n, unified, x);
                }
            }
            addLoopLoadStores(n) {
                const [body_block] = n.blocks();
                const loop_vars = this.addControlFlowLoadStores(body_block);