Method generate_getitems

source/python.js:20403–20451

(meta_val, fx_node, args)

                    individual_output.meta.update(deserialized_metadata);
                };
                const generate_getitems = (meta_val, fx_node, args) => {
                    for (let idx = 0; idx < args.length; idx++) {
                        const arg = args[idx];
                        if (arg instanceof torch._export.serde.schema.TensorArgument ||
                            arg instanceof torch._export.serde.schema.SymIntArgument ||
                            arg instanceof torch._export.serde.schema.SymFloatArgument) {
                            generate_getitem(meta_val, fx_node, arg, idx);
                        } else if (arg instanceof torch._export.serde.schema.Argument) {
                            if (arg.type === 'as_tensor' || arg.type === 'as_sym_int' || arg.type === 'as_sym_float') {
                                generate_getitem(meta_val, fx_node, arg.value, idx);
                            } else if (arg.type === 'as_none') {
                                const individual_output = this.graph.create_node(
                                    'call_function',
                                    operator.getitem,
                                    new builtins.tuple([fx_node, idx]),
                                    null,
                                    'as_none',
                                );
                                meta_val.push(null);
                                individual_output.meta.set('val', null);
                                individual_output.meta.update(deserialized_metadata);
                            } else if (Array.isArray(arg.value)) {
                                const list_output = this.graph.create_node(
                                    'call_function',
                                    operator.getitem,
                                    new builtins.tuple([fx_node, idx]),
                                );
                                meta_val.push([]);
                                generate_getitems(meta_val[meta_val.length - 1], list_output, arg.value);
                                list_output.meta.update(deserialized_metadata);
                                list_output.meta.set('val', meta_val[meta_val.length - 1]);
                            } else {
                                throw new python.Error(`Unsupported node output type '${arg.type}'.`);
                            }
                        } else if (Array.isArray(arg)) {
                            const list_output = this.graph.create_node(
                                'call_function',
                                operator.getitem,
                                new builtins.tuple([fx_node, idx]),
                            );
                            meta_val.push([]);
                            generate_getitems(meta_val[meta_val.length - 1], list_output, arg);
                            list_output.meta.update(deserialized_metadata);
                            list_output.meta.set('val', meta_val[meta_val.length - 1]);
                        } else {
                            throw new python.Error(`Unsupported node output type '${arg}'.`);
                        }
                    }
                };
                const meta_val = [];
                if (serialized_node.outputs.length === 1) {
                    generate_getitems(meta_val, fx_node, serialized_node.outputs[0].value);