{
  "layer_shapes" : {
    "input.95" : {
      "k" : 48,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "x_1" : {
      "k" : 6,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "input.41" : {
      "k" : 24,
      "w" : 104,
      "n" : 1,
      "h" : 78
    },
    "input.115" : {
      "k" : 128,
      "w" : 52,
      "n" : 1,
      "h" : 39
    },
    "x.11_divide" : {
      "k" : 156,
      "w" : 12,
      "n" : 1,
      "h" : 208
    },
    "out_uv_slices" : {
      "k" : 2,
      "w" : 832,
      "n" : 1,
      "h" : 624
    },
    "x.47_sub_mean" : {
      "k" : 156,
      "w" : 24,
      "n" : 1,
      "h" : 208
    },
    "output_tensor" : {
      "k" : 1,
      "w" : 1664,
      "n" : 1,
      "h" : 1248
    },
    "x.17_mean" : {
      "k" : 78,
      "w" : 1,
      "n" : 1,
      "h" : 104
    },
    "x.51" : {
      "k" : 12,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "x.29_epsilon" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "x.5_div_square_sum" : {
      "k" : 156,
      "w" : 1,
      "n" : 1,
      "h" : 208
    },
    "x.35_div_constant" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "x.41_square_sum" : {
      "k" : 156,
      "w" : 1,
      "n" : 1,
      "h" : 208
    },
    "x.17_div_square_sum" : {
      "k" : 78,
      "w" : 1,
      "n" : 1,
      "h" : 104
    },
    "transpose_17" : {
      "k" : 156,
      "w" : 6,
      "n" : 1,
      "h" : 208
    },
    "x.5_sqrt" : {
      "k" : 156,
      "w" : 1,
      "n" : 1,
      "h" : 208
    },
    "x.53" : {
      "k" : 156,
      "w" : 12,
      "n" : 1,
      "h" : 208
    },
    "transpose_15" : {
      "k" : 156,
      "w" : 12,
      "n" : 1,
      "h" : 208
    },
    "x.53_div_square_sum" : {
      "k" : 156,
      "w" : 1,
      "n" : 1,
      "h" : 208
    },
    "x.53_epsilon" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "x.23_mean" : {
      "k" : 78,
      "w" : 1,
      "n" : 1,
      "h" : 104
    },
    "input.27" : {
      "k" : 48,
      "w" : 104,
      "n" : 1,
      "h" : 78
    },
    "transpose_9" : {
      "k" : 39,
      "w" : 24,
      "n" : 1,
      "h" : 52
    },
    "transpose_13" : {
      "k" : 78,
      "w" : 12,
      "n" : 1,
      "h" : 104
    },
    "x.29_square" : {
      "k" : 39,
      "w" : 24,
      "n" : 1,
      "h" : 52
    },
    "x.41" : {
      "k" : 156,
      "w" : 36,
      "n" : 1,
      "h" : 208
    },
    "transpose_5" : {
      "k" : 156,
      "w" : 36,
      "n" : 1,
      "h" : 208
    },
    "transpose_11" : {
      "k" : 78,
      "w" : 24,
      "n" : 1,
      "h" : 104
    },
    "x.23_add_epsilon" : {
      "k" : 78,
      "w" : 1,
      "n" : 1,
      "h" : 104
    },
    "transpose_1" : {
      "k" : 156,
      "w" : 12,
      "n" : 1,
      "h" : 208
    },
    "x.35_div_square_sum" : {
      "k" : 78,
      "w" : 1,
      "n" : 1,
      "h" : 104
    },
    "input.87" : {
      "k" : 12,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "input.153" : {
      "k" : 256,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "input" : {
      "k" : 16,
      "w" : 832,
      "n" : 1,
      "h" : 624
    },
    "x.53_div_constant" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "residual.5" : {
      "k" : 48,
      "w" : 52,
      "n" : 1,
      "h" : 39
    },
    "residual.11" : {
      "k" : 12,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "x.11_beta" : {
      "k" : 1,
      "w" : 12,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "input.63" : {
      "k" : 288,
      "w" : 104,
      "n" : 1,
      "h" : 78
    },
    "x.53_add_epsilon" : {
      "k" : 156,
      "w" : 1,
      "n" : 1,
      "h" : 208
    },
    "x.43" : {
      "k" : 12,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "out_uv_scaled" : {
      "k" : 2,
      "w" : 832,
      "n" : 1,
      "h" : 624
    },
    "x.5_square_sum" : {
      "k" : 156,
      "w" : 1,
      "n" : 1,
      "h" : 208
    },
    "input.93" : {
      "k" : 48,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "x.31" : {
      "k" : 48,
      "w" : 52,
      "n" : 1,
      "h" : 39
    },
    "x.41_div_square_sum" : {
      "k" : 156,
      "w" : 1,
      "n" : 1,
      "h" : 208
    },
    "out_Y_plane" : {
      "k" : 1,
      "w" : 1664,
      "n" : 1,
      "h" : 1248
    },
    "x.47_sqrt" : {
      "k" : 156,
      "w" : 1,
      "n" : 1,
      "h" : 208
    },
    "x.35_square_sum" : {
      "k" : 78,
      "w" : 1,
      "n" : 1,
      "h" : 104
    },
    "input.143" : {
      "k" : 16,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "x.45" : {
      "k" : 24,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "out_y_slices" : {
      "k" : 4,
      "w" : 832,
      "n" : 1,
      "h" : 624
    },
    "x.17_mul" : {
      "k" : 78,
      "w" : 12,
      "n" : 1,
      "h" : 104
    },
    "x.5_beta" : {
      "k" : 1,
      "w" : 6,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "x.35_epsilon" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "x.23_div_square_sum" : {
      "k" : 78,
      "w" : 1,
      "n" : 1,
      "h" : 104
    },
    "x.33" : {
      "k" : 72,
      "w" : 104,
      "n" : 1,
      "h" : 78
    },
    "input.19" : {
      "k" : 12,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "x.47_square" : {
      "k" : 156,
      "w" : 24,
      "n" : 1,
      "h" : 208
    },
    "x.17_add_epsilon" : {
      "k" : 78,
      "w" : 1,
      "n" : 1,
      "h" : 104
    },
    "input.119" : {
      "k" : 128,
      "w" : 52,
      "n" : 1,
      "h" : 39
    },
    "input.133" : {
      "k" : 32,
      "w" : 104,
      "n" : 1,
      "h" : 78
    },
    "x.21" : {
      "k" : 24,
      "w" : 104,
      "n" : 1,
      "h" : 78
    },
    "input.49" : {
      "k" : 96,
      "w" : 52,
      "n" : 1,
      "h" : 39
    },
    "x.53_sqrt" : {
      "k" : 156,
      "w" : 1,
      "n" : 1,
      "h" : 208
    },
    "x.47" : {
      "k" : 156,
      "w" : 24,
      "n" : 1,
      "h" : 208
    },
    "output_y" : {
      "k" : 1,
      "w" : 1664,
      "n" : 1,
      "h" : 1248
    },
    "input.9" : {
      "k" : 12,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "x.23_divide" : {
      "k" : 78,
      "w" : 24,
      "n" : 1,
      "h" : 104
    },
    "x.23_div_constant" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "x.35" : {
      "k" : 78,
      "w" : 72,
      "n" : 1,
      "h" : 104
    },
    "x.53_square_sum" : {
      "k" : 156,
      "w" : 1,
      "n" : 1,
      "h" : 208
    },
    "x.35_sub_mean" : {
      "k" : 78,
      "w" : 72,
      "n" : 1,
      "h" : 104
    },
    "x.41_divide" : {
      "k" : 156,
      "w" : 36,
      "n" : 1,
      "h" : 208
    },
    "out_Y_plane_scaled" : {
      "k" : 1,
      "w" : 1664,
      "n" : 1,
      "h" : 1248
    },
    "x.47_add_epsilon" : {
      "k" : 156,
      "w" : 1,
      "n" : 1,
      "h" : 208
    },
    "x.5_gamma" : {
      "k" : 1,
      "w" : 6,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "input.55" : {
      "k" : 72,
      "w" : 104,
      "n" : 1,
      "h" : 78
    },
    "x.35_mul" : {
      "k" : 78,
      "w" : 72,
      "n" : 1,
      "h" : 104
    },
    "516" : {
      "k" : 32,
      "w" : 104,
      "n" : 1,
      "h" : 78
    },
    "x.23" : {
      "k" : 78,
      "w" : 24,
      "n" : 1,
      "h" : 104
    },
    "input.123" : {
      "k" : 64,
      "w" : 52,
      "n" : 1,
      "h" : 39
    },
    "input.85" : {
      "k" : 96,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "x.49" : {
      "k" : 12,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "input.31" : {
      "k" : 24,
      "w" : 104,
      "n" : 1,
      "h" : 78
    },
    "cropped_y" : {
      "k" : 1,
      "w" : 416,
      "n" : 1,
      "h" : 312
    },
    "x.11_mean" : {
      "k" : 156,
      "w" : 1,
      "n" : 1,
      "h" : 208
    },
    "x.29_square_sum" : {
      "k" : 39,
      "w" : 1,
      "n" : 1,
      "h" : 52
    },
    "x.17_gamma" : {
      "k" : 1,
      "w" : 12,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "x.5_divide" : {
      "k" : 156,
      "w" : 6,
      "n" : 1,
      "h" : 208
    },
    "x.11" : {
      "k" : 156,
      "w" : 12,
      "n" : 1,
      "h" : 208
    },
    "x.37" : {
      "k" : 24,
      "w" : 104,
      "n" : 1,
      "h" : 78
    },
    "input.61" : {
      "k" : 288,
      "w" : 104,
      "n" : 1,
      "h" : 78
    },
    "x4u" : {
      "k" : 48,
      "w" : 104,
      "n" : 1,
      "h" : 78
    },
    "output_uv" : {
      "k" : 2,
      "w" : 832,
      "n" : 1,
      "h" : 624
    },
    "x.29_sqrt" : {
      "k" : 39,
      "w" : 1,
      "n" : 1,
      "h" : 52
    },
    "transpose_6" : {
      "k" : 72,
      "w" : 104,
      "n" : 1,
      "h" : 78
    },
    "x.25" : {
      "k" : 24,
      "w" : 104,
      "n" : 1,
      "h" : 78
    },
    "x.41_div_constant" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "transpose_2" : {
      "k" : 24,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "var_573" : {
      "k" : 6,
      "w" : 832,
      "n" : 1,
      "h" : 624
    },
    "x.53_mul" : {
      "k" : 156,
      "w" : 12,
      "n" : 1,
      "h" : 208
    },
    "x.29_gamma" : {
      "k" : 1,
      "w" : 24,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "x.11_gamma" : {
      "k" : 1,
      "w" : 12,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "x.17_epsilon" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "x.13" : {
      "k" : 12,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "residual.7" : {
      "k" : 24,
      "w" : 104,
      "n" : 1,
      "h" : 78
    },
    "x.39" : {
      "k" : 36,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "x.5_sub_mean" : {
      "k" : 156,
      "w" : 6,
      "n" : 1,
      "h" : 208
    },
    "x.3" : {
      "k" : 6,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "x.17_divide" : {
      "k" : 78,
      "w" : 12,
      "n" : 1,
      "h" : 104
    },
    "x.27" : {
      "k" : 24,
      "w" : 52,
      "n" : 1,
      "h" : 39
    },
    "input.17" : {
      "k" : 48,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "x.11_square" : {
      "k" : 156,
      "w" : 12,
      "n" : 1,
      "h" : 208
    },
    "x.35_sqrt" : {
      "k" : 78,
      "w" : 1,
      "n" : 1,
      "h" : 104
    },
    "x.23_gamma" : {
      "k" : 1,
      "w" : 24,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "x.41_epsilon" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "x3d" : {
      "k" : 12,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "x.5" : {
      "k" : 156,
      "w" : 6,
      "n" : 1,
      "h" : 208
    },
    "input.103" : {
      "k" : 16,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "x.15" : {
      "k" : 12,
      "w" : 104,
      "n" : 1,
      "h" : 78
    },
    "input.147" : {
      "k" : 32,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "x.11_div_square_sum" : {
      "k" : 156,
      "w" : 1,
      "n" : 1,
      "h" : 208
    },
    "x.47_beta" : {
      "k" : 1,
      "w" : 24,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "input.5" : {
      "k" : 24,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "input.77" : {
      "k" : 24,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "x.7" : {
      "k" : 12,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "x.11_add_epsilon" : {
      "k" : 156,
      "w" : 1,
      "n" : 1,
      "h" : 208
    },
    "x.29" : {
      "k" : 39,
      "w" : 24,
      "n" : 1,
      "h" : 52
    },
    "x.35_gamma" : {
      "k" : 1,
      "w" : 72,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "input.53" : {
      "k" : 48,
      "w" : 52,
      "n" : 1,
      "h" : 39
    },
    "x.5_mean" : {
      "k" : 156,
      "w" : 1,
      "n" : 1,
      "h" : 208
    },
    "x.9" : {
      "k" : 12,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "x.41_sqrt" : {
      "k" : 156,
      "w" : 1,
      "n" : 1,
      "h" : 208
    },
    "x.17" : {
      "k" : 78,
      "w" : 12,
      "n" : 1,
      "h" : 104
    },
    "input.83" : {
      "k" : 96,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "input.137" : {
      "k" : 32,
      "w" : 104,
      "n" : 1,
      "h" : 78
    },
    "input.151" : {
      "k" : 32,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "x.11_square_sum" : {
      "k" : 156,
      "w" : 1,
      "n" : 1,
      "h" : 208
    },
    "x.41_add_epsilon" : {
      "k" : 156,
      "w" : 1,
      "n" : 1,
      "h" : 208
    },
    "x.47_gamma" : {
      "k" : 1,
      "w" : 24,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "x.53_beta" : {
      "k" : 1,
      "w" : 12,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "x.47_square_sum" : {
      "k" : 156,
      "w" : 1,
      "n" : 1,
      "h" : 208
    },
    "x.35_divide" : {
      "k" : 78,
      "w" : 72,
      "n" : 1,
      "h" : 104
    },
    "x.19" : {
      "k" : 24,
      "w" : 104,
      "n" : 1,
      "h" : 78
    },
    "495" : {
      "k" : 64,
      "w" : 52,
      "n" : 1,
      "h" : 39
    },
    "x.53_divide" : {
      "k" : 156,
      "w" : 12,
      "n" : 1,
      "h" : 208
    },
    "x.41_gamma" : {
      "k" : 1,
      "w" : 36,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "input.127" : {
      "k" : 64,
      "w" : 52,
      "n" : 1,
      "h" : 39
    },
    "input.141" : {
      "k" : 16,
      "w" : 104,
      "n" : 1,
      "h" : 78
    },
    "x.23_sub_mean" : {
      "k" : 78,
      "w" : 24,
      "n" : 1,
      "h" : 104
    },
    "transpose_16" : {
      "k" : 6,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "x.23_epsilon" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "input.39" : {
      "k" : 96,
      "w" : 104,
      "n" : 1,
      "h" : 78
    },
    "x.17_sqrt" : {
      "k" : 78,
      "w" : 1,
      "n" : 1,
      "h" : 104
    },
    "transpose_14" : {
      "k" : 12,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "transpose_7" : {
      "k" : 78,
      "w" : 72,
      "n" : 1,
      "h" : 104
    },
    "transpose_12" : {
      "k" : 12,
      "w" : 104,
      "n" : 1,
      "h" : 78
    },
    "input.15" : {
      "k" : 48,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "x.53_gamma" : {
      "k" : 1,
      "w" : 12,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "transpose_3" : {
      "k" : 156,
      "w" : 24,
      "n" : 1,
      "h" : 208
    },
    "x.29_beta" : {
      "k" : 1,
      "w" : 24,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "transpose_10" : {
      "k" : 24,
      "w" : 104,
      "n" : 1,
      "h" : 78
    },
    "input.99" : {
      "k" : 6,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "x.11_div_constant" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "x.47_mean" : {
      "k" : 156,
      "w" : 1,
      "n" : 1,
      "h" : 208
    },
    "cropped_uv" : {
      "k" : 2,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "input.131" : {
      "k" : 32,
      "w" : 52,
      "n" : 1,
      "h" : 39
    },
    "residual.9" : {
      "k" : 12,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "x.23_mul" : {
      "k" : 78,
      "w" : 24,
      "n" : 1,
      "h" : 104
    },
    "residual.1" : {
      "k" : 12,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "input.75" : {
      "k" : 144,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "x.35_add_epsilon" : {
      "k" : 78,
      "w" : 1,
      "n" : 1,
      "h" : 104
    },
    "input.21" : {
      "k" : 12,
      "w" : 104,
      "n" : 1,
      "h" : 78
    },
    "537" : {
      "k" : 16,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "x.23_sqrt" : {
      "k" : 78,
      "w" : 1,
      "n" : 1,
      "h" : 104
    },
    "input.51" : {
      "k" : 96,
      "w" : 52,
      "n" : 1,
      "h" : 39
    },
    "x3u" : {
      "k" : 24,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "out_uv_unbiased" : {
      "k" : 2,
      "w" : 832,
      "n" : 1,
      "h" : 624
    },
    "x.35_beta" : {
      "k" : 1,
      "w" : 72,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "x.29_divide" : {
      "k" : 39,
      "w" : 24,
      "n" : 1,
      "h" : 52
    },
    "input.107" : {
      "k" : 32,
      "w" : 104,
      "n" : 1,
      "h" : 78
    },
    "x.23_square" : {
      "k" : 78,
      "w" : 24,
      "n" : 1,
      "h" : 104
    },
    "input_uv" : {
      "k" : 2,
      "w" : 832,
      "n" : 1,
      "h" : 624
    },
    "x.53_mean" : {
      "k" : 156,
      "w" : 1,
      "n" : 1,
      "h" : 208
    },
    "x.53_sub_mean" : {
      "k" : 156,
      "w" : 12,
      "n" : 1,
      "h" : 208
    },
    "x.41_square" : {
      "k" : 156,
      "w" : 36,
      "n" : 1,
      "h" : 208
    },
    "x.5_div_constant" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "x.29_sub_mean" : {
      "k" : 39,
      "w" : 24,
      "n" : 1,
      "h" : 52
    },
    "x.41_mul" : {
      "k" : 156,
      "w" : 36,
      "n" : 1,
      "h" : 208
    },
    "x.29_div_constant" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "input.111" : {
      "k" : 64,
      "w" : 52,
      "n" : 1,
      "h" : 39
    },
    "x.41_beta" : {
      "k" : 1,
      "w" : 36,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "input.37" : {
      "k" : 96,
      "w" : 104,
      "n" : 1,
      "h" : 78
    },
    "x.29_add_epsilon" : {
      "k" : 39,
      "w" : 1,
      "n" : 1,
      "h" : 52
    },
    "x" : {
      "k" : 12,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "input.67" : {
      "k" : 36,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "uv_biased" : {
      "k" : 2,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "x.29_mean" : {
      "k" : 39,
      "w" : 1,
      "n" : 1,
      "h" : 52
    },
    "x.5_add_epsilon" : {
      "k" : 156,
      "w" : 1,
      "n" : 1,
      "h" : 208
    },
    "input.97" : {
      "k" : 12,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "input.43" : {
      "k" : 24,
      "w" : 52,
      "n" : 1,
      "h" : 39
    },
    "x.5_mul" : {
      "k" : 156,
      "w" : 6,
      "n" : 1,
      "h" : 208
    },
    "x.47_divide" : {
      "k" : 156,
      "w" : 24,
      "n" : 1,
      "h" : 208
    },
    "x.47_div_constant" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "420" : {
      "k" : 6,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "input.73" : {
      "k" : 144,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "x.17_square" : {
      "k" : 78,
      "w" : 12,
      "n" : 1,
      "h" : 104
    },
    "input_y" : {
      "k" : 1,
      "w" : 1664,
      "n" : 1,
      "h" : 1248
    },
    "uv_scaled" : {
      "k" : 2,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "x.17_beta" : {
      "k" : 1,
      "w" : 12,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "transpose_8" : {
      "k" : 24,
      "w" : 52,
      "n" : 1,
      "h" : 39
    },
    "x.11_sub_mean" : {
      "k" : 156,
      "w" : 12,
      "n" : 1,
      "h" : 208
    },
    "x.5_square" : {
      "k" : 156,
      "w" : 6,
      "n" : 1,
      "h" : 208
    },
    "x.23_square_sum" : {
      "k" : 78,
      "w" : 1,
      "n" : 1,
      "h" : 104
    },
    "422" : {
      "k" : 6,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "x.35_mean" : {
      "k" : 78,
      "w" : 1,
      "n" : 1,
      "h" : 104
    },
    "y_scaled" : {
      "k" : 1,
      "w" : 416,
      "n" : 1,
      "h" : 312
    },
    "transpose_4" : {
      "k" : 36,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "input.7" : {
      "k" : 24,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "x.29_mul" : {
      "k" : 39,
      "w" : 24,
      "n" : 1,
      "h" : 52
    },
    "transpose_0" : {
      "k" : 12,
      "w" : 208,
      "n" : 1,
      "h" : 156
    },
    "x.5_epsilon" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "x.17_sub_mean" : {
      "k" : 78,
      "w" : 12,
      "n" : 1,
      "h" : 104
    },
    "x.47_epsilon" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "x.11_sqrt" : {
      "k" : 156,
      "w" : 1,
      "n" : 1,
      "h" : 208
    },
    "residual.3" : {
      "k" : 24,
      "w" : 104,
      "n" : 1,
      "h" : 78
    },
    "input.29" : {
      "k" : 48,
      "w" : 104,
      "n" : 1,
      "h" : 78
    },
    "x.47_div_square_sum" : {
      "k" : 156,
      "w" : 1,
      "n" : 1,
      "h" : 208
    },
    "x.23_beta" : {
      "k" : 1,
      "w" : 24,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "x.11_epsilon" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "x.41_mean" : {
      "k" : 156,
      "w" : 1,
      "n" : 1,
      "h" : 208
    },
    "x.47_mul" : {
      "k" : 156,
      "w" : 24,
      "n" : 1,
      "h" : 208
    },
    "input_tensor" : {
      "k" : 1,
      "w" : 1664,
      "n" : 1,
      "_rank" : 4,
      "h" : 1248
    },
    "x.29_div_square_sum" : {
      "k" : 39,
      "w" : 1,
      "n" : 1,
      "h" : 52
    },
    "x.35_square" : {
      "k" : 78,
      "w" : 72,
      "n" : 1,
      "h" : 104
    },
    "x.17_div_constant" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "input.65" : {
      "k" : 24,
      "w" : 104,
      "n" : 1,
      "h" : 78
    },
    "x.53_square" : {
      "k" : 156,
      "w" : 12,
      "n" : 1,
      "h" : 208
    },
    "x.41_sub_mean" : {
      "k" : 156,
      "w" : 36,
      "n" : 1,
      "h" : 208
    },
    "x.17_square_sum" : {
      "k" : 78,
      "w" : 1,
      "n" : 1,
      "h" : 104
    },
    "x.11_mul" : {
      "k" : 156,
      "w" : 12,
      "n" : 1,
      "h" : 208
    },
    "y_4_channel" : {
      "k" : 4,
      "w" : 208,
      "n" : 1,
      "h" : 156
    }
  }
}