{"0": {"name": "x", "type": "input"}, "1": {"name": "stem_0", "predecessor_indices": [0], "is_atomic": true, "type": "inner", "orbit": "orbit_0", "module_index": 0}, "2": {"name": "stem_1", "predecessor_indices": [1], "is_atomic": true, "type": "inner", "orbit": "orbit_0", "module_index": 1}, "3": {"name": "stages_0_blocks_0_conv_dw", "predecessor_indices": [2], "is_atomic": true, "type": "inner", "orbit": "orbit_0", "module_index": 2}, "4": {"name": "stages_0_blocks_0_norm", "predecessor_indices": [3], "is_atomic": true, "type": "inner", "orbit": "orbit_0", "module_index": 3}, "5": {"name": "orbit_0_mask_0", "predecessor_indices": [4], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 4}, "6": {"name": "stages_0_blocks_0_mlp_fc1", "predecessor_indices": [5], "is_atomic": true, "type": "inner", "orbit": "orbit_1", "module_index": 5}, "7": {"name": "gelu", "predecessor_indices": [6], "is_atomic": true, "type": "inner", "orbit": "orbit_1", "module_index": 6}, "8": {"name": "stages_0_blocks_0_mlp_drop1", "predecessor_indices": [7], "is_atomic": true, "type": "inner", "orbit": "orbit_1", "module_index": 7}, "9": {"name": "stages_0_blocks_0_mlp_grn", "predecessor_indices": [8], "is_atomic": true, "type": "inner", "orbit": "orbit_1", "module_index": 8}, "10": {"name": "orbit_1_mask_0", "predecessor_indices": [9], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 9}, "11": {"name": "stages_0_blocks_0_mlp_fc2", "predecessor_indices": [10], "is_atomic": true, "type": "inner", "orbit": "orbit_0", "module_index": 10}, "12": {"name": "stages_0_blocks_0_mlp_drop2", "predecessor_indices": [11], "is_atomic": true, "type": "inner", "orbit": "orbit_0", "module_index": 11}, "13": {"name": "add", "predecessor_indices": [12, 2], "is_atomic": true, "type": "inner", "orbit": "orbit_0", "module_index": 12}, "14": {"name": "stages_0_blocks_1_conv_dw", "predecessor_indices": [13], "is_atomic": true, "type": "inner", "orbit": "orbit_0", "module_index": 13}, "15": {"name": "stages_0_blocks_1_norm", "predecessor_indices": [14], "is_atomic": true, "type": "inner", "orbit": "orbit_0", "module_index": 14}, "16": {"name": "orbit_0_mask_1", "predecessor_indices": [15], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 15}, "17": {"name": "stages_0_blocks_1_mlp_fc1", "predecessor_indices": [16], "is_atomic": true, "type": "inner", "orbit": "orbit_2", "module_index": 16}, "18": {"name": "gelu_1", "predecessor_indices": [17], "is_atomic": true, "type": "inner", "orbit": "orbit_2", "module_index": 17}, "19": {"name": "stages_0_blocks_1_mlp_drop1", "predecessor_indices": [18], "is_atomic": true, "type": "inner", "orbit": "orbit_2", "module_index": 18}, "20": {"name": "stages_0_blocks_1_mlp_grn", "predecessor_indices": [19], "is_atomic": true, "type": "inner", "orbit": "orbit_2", "module_index": 19}, "21": {"name": "orbit_2_mask_0", "predecessor_indices": [20], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 20}, "22": {"name": "stages_0_blocks_1_mlp_fc2", "predecessor_indices": [21], "is_atomic": true, "type": "inner", "orbit": "orbit_0", "module_index": 21}, "23": {"name": "stages_0_blocks_1_mlp_drop2", "predecessor_indices": [22], "is_atomic": true, "type": "inner", "orbit": "orbit_0", "module_index": 22}, "24": {"name": "add_1", "predecessor_indices": [23, 13], "is_atomic": true, "type": "inner", "orbit": "orbit_0", "module_index": 23}, "25": {"name": "stages_1_downsample_0", "predecessor_indices": [24], "is_atomic": true, "type": "inner", "orbit": "orbit_0", "module_index": 24}, "26": {"name": "orbit_0_mask_2", "predecessor_indices": [25], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 25}, "27": {"name": "stages_1_downsample_1", "predecessor_indices": [26], "is_atomic": true, "type": "inner", "orbit": "orbit_3", "module_index": 26}, "28": {"name": "stages_1_blocks_0_conv_dw", "predecessor_indices": [27], "is_atomic": true, "type": "inner", "orbit": "orbit_3", "module_index": 27}, "29": {"name": "stages_1_blocks_0_norm", "predecessor_indices": [28], "is_atomic": true, "type": "inner", "orbit": "orbit_3", "module_index": 28}, "30": {"name": "orbit_3_mask_0", "predecessor_indices": [29], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 29}, "31": {"name": "stages_1_blocks_0_mlp_fc1", "predecessor_indices": [30], "is_atomic": true, "type": "inner", "orbit": "orbit_4", "module_index": 30}, "32": {"name": "gelu_2", "predecessor_indices": [31], "is_atomic": true, "type": "inner", "orbit": "orbit_4", "module_index": 31}, "33": {"name": "stages_1_blocks_0_mlp_drop1", "predecessor_indices": [32], "is_atomic": true, "type": "inner", "orbit": "orbit_4", "module_index": 32}, "34": {"name": "stages_1_blocks_0_mlp_grn", "predecessor_indices": [33], "is_atomic": true, "type": "inner", "orbit": "orbit_4", "module_index": 33}, "35": {"name": "orbit_4_mask_0", "predecessor_indices": [34], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 34}, "36": {"name": "stages_1_blocks_0_mlp_fc2", "predecessor_indices": [35], "is_atomic": true, "type": "inner", "orbit": "orbit_3", "module_index": 35}, "37": {"name": "stages_1_blocks_0_mlp_drop2", "predecessor_indices": [36], "is_atomic": true, "type": "inner", "orbit": "orbit_3", "module_index": 36}, "38": {"name": "add_2", "predecessor_indices": [37, 27], "is_atomic": true, "type": "inner", "orbit": "orbit_3", "module_index": 37}, "39": {"name": "stages_1_blocks_1_conv_dw", "predecessor_indices": [38], "is_atomic": true, "type": "inner", "orbit": "orbit_3", "module_index": 38}, "40": {"name": "stages_1_blocks_1_norm", "predecessor_indices": [39], "is_atomic": true, "type": "inner", "orbit": "orbit_3", "module_index": 39}, "41": {"name": "orbit_3_mask_1", "predecessor_indices": [40], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 40}, "42": {"name": "stages_1_blocks_1_mlp_fc1", "predecessor_indices": [41], "is_atomic": true, "type": "inner", "orbit": "orbit_5", "module_index": 41}, "43": {"name": "gelu_3", "predecessor_indices": [42], "is_atomic": true, "type": "inner", "orbit": "orbit_5", "module_index": 42}, "44": {"name": "stages_1_blocks_1_mlp_drop1", "predecessor_indices": [43], "is_atomic": true, "type": "inner", "orbit": "orbit_5", "module_index": 43}, "45": {"name": "stages_1_blocks_1_mlp_grn", "predecessor_indices": [44], "is_atomic": true, "type": "inner", "orbit": "orbit_5", "module_index": 44}, "46": {"name": "orbit_5_mask_0", "predecessor_indices": [45], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 45}, "47": {"name": "stages_1_blocks_1_mlp_fc2", "predecessor_indices": [46], "is_atomic": true, "type": "inner", "orbit": "orbit_3", "module_index": 46}, "48": {"name": "stages_1_blocks_1_mlp_drop2", "predecessor_indices": [47], "is_atomic": true, "type": "inner", "orbit": "orbit_3", "module_index": 47}, "49": {"name": "add_3", "predecessor_indices": [48, 38], "is_atomic": true, "type": "inner", "orbit": "orbit_3", "module_index": 48}, "50": {"name": "stages_2_downsample_0", "predecessor_indices": [49], "is_atomic": true, "type": "inner", "orbit": "orbit_3", "module_index": 49}, "51": {"name": "orbit_3_mask_2", "predecessor_indices": [50], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 50}, "52": {"name": "stages_2_downsample_1", "predecessor_indices": [51], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 51}, "53": {"name": "stages_2_blocks_0_conv_dw", "predecessor_indices": [52], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 52}, "54": {"name": "stages_2_blocks_0_norm", "predecessor_indices": [53], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 53}, "55": {"name": "orbit_6_mask_0", "predecessor_indices": [54], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 54}, "56": {"name": "stages_2_blocks_0_mlp_fc1", "predecessor_indices": [55], "is_atomic": true, "type": "inner", "orbit": "orbit_7", "module_index": 55}, "57": {"name": "gelu_4", "predecessor_indices": [56], "is_atomic": true, "type": "inner", "orbit": "orbit_7", "module_index": 56}, "58": {"name": "stages_2_blocks_0_mlp_drop1", "predecessor_indices": [57], "is_atomic": true, "type": "inner", "orbit": "orbit_7", "module_index": 57}, "59": {"name": "stages_2_blocks_0_mlp_grn", "predecessor_indices": [58], "is_atomic": true, "type": "inner", "orbit": "orbit_7", "module_index": 58}, "60": {"name": "orbit_7_mask_0", "predecessor_indices": [59], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 59}, "61": {"name": "stages_2_blocks_0_mlp_fc2", "predecessor_indices": [60], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 60}, "62": {"name": "stages_2_blocks_0_mlp_drop2", "predecessor_indices": [61], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 61}, "63": {"name": "add_4", "predecessor_indices": [62, 52], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 62}, "64": {"name": "stages_2_blocks_1_conv_dw", "predecessor_indices": [63], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 63}, "65": {"name": "stages_2_blocks_1_norm", "predecessor_indices": [64], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 64}, "66": {"name": "orbit_6_mask_1", "predecessor_indices": [65], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 65}, "67": {"name": "stages_2_blocks_1_mlp_fc1", "predecessor_indices": [66], "is_atomic": true, "type": "inner", "orbit": "orbit_8", "module_index": 66}, "68": {"name": "gelu_5", "predecessor_indices": [67], "is_atomic": true, "type": "inner", "orbit": "orbit_8", "module_index": 67}, "69": {"name": "stages_2_blocks_1_mlp_drop1", "predecessor_indices": [68], "is_atomic": true, "type": "inner", "orbit": "orbit_8", "module_index": 68}, "70": {"name": "stages_2_blocks_1_mlp_grn", "predecessor_indices": [69], "is_atomic": true, "type": "inner", "orbit": "orbit_8", "module_index": 69}, "71": {"name": "orbit_8_mask_0", "predecessor_indices": [70], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 70}, "72": {"name": "stages_2_blocks_1_mlp_fc2", "predecessor_indices": [71], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 71}, "73": {"name": "stages_2_blocks_1_mlp_drop2", "predecessor_indices": [72], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 72}, "74": {"name": "add_5", "predecessor_indices": [73, 63], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 73}, "75": {"name": "stages_2_blocks_2_conv_dw", "predecessor_indices": [74], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 74}, "76": {"name": "stages_2_blocks_2_norm", "predecessor_indices": [75], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 75}, "77": {"name": "orbit_6_mask_2", "predecessor_indices": [76], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 76}, "78": {"name": "stages_2_blocks_2_mlp_fc1", "predecessor_indices": [77], "is_atomic": true, "type": "inner", "orbit": "orbit_9", "module_index": 77}, "79": {"name": "gelu_6", "predecessor_indices": [78], "is_atomic": true, "type": "inner", "orbit": "orbit_9", "module_index": 78}, "80": {"name": "stages_2_blocks_2_mlp_drop1", "predecessor_indices": [79], "is_atomic": true, "type": "inner", "orbit": "orbit_9", "module_index": 79}, "81": {"name": "stages_2_blocks_2_mlp_grn", "predecessor_indices": [80], "is_atomic": true, "type": "inner", "orbit": "orbit_9", "module_index": 80}, "82": {"name": "orbit_9_mask_0", "predecessor_indices": [81], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 81}, "83": {"name": "stages_2_blocks_2_mlp_fc2", "predecessor_indices": [82], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 82}, "84": {"name": "stages_2_blocks_2_mlp_drop2", "predecessor_indices": [83], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 83}, "85": {"name": "add_6", "predecessor_indices": [84, 74], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 84}, "86": {"name": "stages_2_blocks_3_conv_dw", "predecessor_indices": [85], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 85}, "87": {"name": "stages_2_blocks_3_norm", "predecessor_indices": [86], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 86}, "88": {"name": "orbit_6_mask_3", "predecessor_indices": [87], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 87}, "89": {"name": "stages_2_blocks_3_mlp_fc1", "predecessor_indices": [88], "is_atomic": true, "type": "inner", "orbit": "orbit_10", "module_index": 88}, "90": {"name": "gelu_7", "predecessor_indices": [89], "is_atomic": true, "type": "inner", "orbit": "orbit_10", "module_index": 89}, "91": {"name": "stages_2_blocks_3_mlp_drop1", "predecessor_indices": [90], "is_atomic": true, "type": "inner", "orbit": "orbit_10", "module_index": 90}, "92": {"name": "stages_2_blocks_3_mlp_grn", "predecessor_indices": [91], "is_atomic": true, "type": "inner", "orbit": "orbit_10", "module_index": 91}, "93": {"name": "orbit_10_mask_0", "predecessor_indices": [92], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 92}, "94": {"name": "stages_2_blocks_3_mlp_fc2", "predecessor_indices": [93], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 93}, "95": {"name": "stages_2_blocks_3_mlp_drop2", "predecessor_indices": [94], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 94}, "96": {"name": "add_7", "predecessor_indices": [95, 85], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 95}, "97": {"name": "stages_2_blocks_4_conv_dw", "predecessor_indices": [96], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 96}, "98": {"name": "stages_2_blocks_4_norm", "predecessor_indices": [97], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 97}, "99": {"name": "orbit_6_mask_4", "predecessor_indices": [98], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 98}, "100": {"name": "stages_2_blocks_4_mlp_fc1", "predecessor_indices": [99], "is_atomic": true, "type": "inner", "orbit": "orbit_11", "module_index": 99}, "101": {"name": "gelu_8", "predecessor_indices": [100], "is_atomic": true, "type": "inner", "orbit": "orbit_11", "module_index": 100}, "102": {"name": "stages_2_blocks_4_mlp_drop1", "predecessor_indices": [101], "is_atomic": true, "type": "inner", "orbit": "orbit_11", "module_index": 101}, "103": {"name": "stages_2_blocks_4_mlp_grn", "predecessor_indices": [102], "is_atomic": true, "type": "inner", "orbit": "orbit_11", "module_index": 102}, "104": {"name": "orbit_11_mask_0", "predecessor_indices": [103], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 103}, "105": {"name": "stages_2_blocks_4_mlp_fc2", "predecessor_indices": [104], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 104}, "106": {"name": "stages_2_blocks_4_mlp_drop2", "predecessor_indices": [105], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 105}, "107": {"name": "add_8", "predecessor_indices": [106, 96], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 106}, "108": {"name": "stages_2_blocks_5_conv_dw", "predecessor_indices": [107], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 107}, "109": {"name": "stages_2_blocks_5_norm", "predecessor_indices": [108], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 108}, "110": {"name": "orbit_6_mask_5", "predecessor_indices": [109], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 109}, "111": {"name": "stages_2_blocks_5_mlp_fc1", "predecessor_indices": [110], "is_atomic": true, "type": "inner", "orbit": "orbit_12", "module_index": 110}, "112": {"name": "gelu_9", "predecessor_indices": [111], "is_atomic": true, "type": "inner", "orbit": "orbit_12", "module_index": 111}, "113": {"name": "stages_2_blocks_5_mlp_drop1", "predecessor_indices": [112], "is_atomic": true, "type": "inner", "orbit": "orbit_12", "module_index": 112}, "114": {"name": "stages_2_blocks_5_mlp_grn", "predecessor_indices": [113], "is_atomic": true, "type": "inner", "orbit": "orbit_12", "module_index": 113}, "115": {"name": "orbit_12_mask_0", "predecessor_indices": [114], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 114}, "116": {"name": "stages_2_blocks_5_mlp_fc2", "predecessor_indices": [115], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 115}, "117": {"name": "stages_2_blocks_5_mlp_drop2", "predecessor_indices": [116], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 116}, "118": {"name": "add_9", "predecessor_indices": [117, 107], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 117}, "119": {"name": "stages_2_blocks_6_conv_dw", "predecessor_indices": [118], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 118}, "120": {"name": "stages_2_blocks_6_norm", "predecessor_indices": [119], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 119}, "121": {"name": "orbit_6_mask_6", "predecessor_indices": [120], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 120}, "122": {"name": "stages_2_blocks_6_mlp_fc1", "predecessor_indices": [121], "is_atomic": true, "type": "inner", "orbit": "orbit_13", "module_index": 121}, "123": {"name": "gelu_10", "predecessor_indices": [122], "is_atomic": true, "type": "inner", "orbit": "orbit_13", "module_index": 122}, "124": {"name": "stages_2_blocks_6_mlp_drop1", "predecessor_indices": [123], "is_atomic": true, "type": "inner", "orbit": "orbit_13", "module_index": 123}, "125": {"name": "stages_2_blocks_6_mlp_grn", "predecessor_indices": [124], "is_atomic": true, "type": "inner", "orbit": "orbit_13", "module_index": 124}, "126": {"name": "orbit_13_mask_0", "predecessor_indices": [125], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 125}, "127": {"name": "stages_2_blocks_6_mlp_fc2", "predecessor_indices": [126], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 126}, "128": {"name": "stages_2_blocks_6_mlp_drop2", "predecessor_indices": [127], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 127}, "129": {"name": "add_10", "predecessor_indices": [128, 118], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 128}, "130": {"name": "stages_2_blocks_7_conv_dw", "predecessor_indices": [129], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 129}, "131": {"name": "stages_2_blocks_7_norm", "predecessor_indices": [130], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 130}, "132": {"name": "orbit_6_mask_7", "predecessor_indices": [131], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 131}, "133": {"name": "stages_2_blocks_7_mlp_fc1", "predecessor_indices": [132], "is_atomic": true, "type": "inner", "orbit": "orbit_14", "module_index": 132}, "134": {"name": "gelu_11", "predecessor_indices": [133], "is_atomic": true, "type": "inner", "orbit": "orbit_14", "module_index": 133}, "135": {"name": "stages_2_blocks_7_mlp_drop1", "predecessor_indices": [134], "is_atomic": true, "type": "inner", "orbit": "orbit_14", "module_index": 134}, "136": {"name": "stages_2_blocks_7_mlp_grn", "predecessor_indices": [135], "is_atomic": true, "type": "inner", "orbit": "orbit_14", "module_index": 135}, "137": {"name": "orbit_14_mask_0", "predecessor_indices": [136], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 136}, "138": {"name": "stages_2_blocks_7_mlp_fc2", "predecessor_indices": [137], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 137}, "139": {"name": "stages_2_blocks_7_mlp_drop2", "predecessor_indices": [138], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 138}, "140": {"name": "add_11", "predecessor_indices": [139, 129], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 139}, "141": {"name": "stages_3_downsample_0", "predecessor_indices": [140], "is_atomic": true, "type": "inner", "orbit": "orbit_6", "module_index": 140}, "142": {"name": "orbit_6_mask_8", "predecessor_indices": [141], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 141}, "143": {"name": "stages_3_downsample_1", "predecessor_indices": [142], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 142}, "144": {"name": "stages_3_blocks_0_conv_dw", "predecessor_indices": [143], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 143}, "145": {"name": "stages_3_blocks_0_norm", "predecessor_indices": [144], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 144}, "146": {"name": "stages_3_blocks_0_mlp_fc1", "predecessor_indices": [145], "is_atomic": true, "type": "inner", "orbit": "orbit_15", "module_index": 145}, "147": {"name": "gelu_12", "predecessor_indices": [146], "is_atomic": true, "type": "inner", "orbit": "orbit_15", "module_index": 146}, "148": {"name": "stages_3_blocks_0_mlp_drop1", "predecessor_indices": [147], "is_atomic": true, "type": "inner", "orbit": "orbit_15", "module_index": 147}, "149": {"name": "stages_3_blocks_0_mlp_grn", "predecessor_indices": [148], "is_atomic": true, "type": "inner", "orbit": "orbit_15", "module_index": 148}, "150": {"name": "orbit_15_mask_0", "predecessor_indices": [149], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 149}, "151": {"name": "stages_3_blocks_0_mlp_fc2", "predecessor_indices": [150], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 150}, "152": {"name": "stages_3_blocks_0_mlp_drop2", "predecessor_indices": [151], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 151}, "153": {"name": "add_12", "predecessor_indices": [152, 143], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 152}, "154": {"name": "stages_3_blocks_1_conv_dw", "predecessor_indices": [153], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 153}, "155": {"name": "stages_3_blocks_1_norm", "predecessor_indices": [154], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 154}, "156": {"name": "stages_3_blocks_1_mlp_fc1", "predecessor_indices": [155], "is_atomic": true, "type": "inner", "orbit": "orbit_16", "module_index": 155}, "157": {"name": "gelu_13", "predecessor_indices": [156], "is_atomic": true, "type": "inner", "orbit": "orbit_16", "module_index": 156}, "158": {"name": "stages_3_blocks_1_mlp_drop1", "predecessor_indices": [157], "is_atomic": true, "type": "inner", "orbit": "orbit_16", "module_index": 157}, "159": {"name": "stages_3_blocks_1_mlp_grn", "predecessor_indices": [158], "is_atomic": true, "type": "inner", "orbit": "orbit_16", "module_index": 158}, "160": {"name": "orbit_16_mask_0", "predecessor_indices": [159], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 159}, "161": {"name": "stages_3_blocks_1_mlp_fc2", "predecessor_indices": [160], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 160}, "162": {"name": "stages_3_blocks_1_mlp_drop2", "predecessor_indices": [161], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 161}, "163": {"name": "add_13", "predecessor_indices": [162, 153], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 162}, "164": {"name": "head_global_pool_pool", "predecessor_indices": [163], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 163}, "165": {"name": "head_norm", "predecessor_indices": [164], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 164}, "166": {"name": "head_flatten", "predecessor_indices": [165], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 165}, "167": {"name": "head_drop", "predecessor_indices": [166], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 166}, "168": {"name": "head_fc", "predecessor_indices": [167], "is_atomic": true, "type": "inner", "orbit": null, "module_index": 167}, "name": "ConvNeXt", "class": "DagModule", "output_index": 168}