{
  "layer_shapes" : {
    "117" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "69" : {
      "k" : 1,
      "w" : 10,
      "n" : 1,
      "_rank" : 3,
      "h" : 1
    },
    "Q" : {
      "k" : 10,
      "w" : 40,
      "n" : 1,
      "_rank" : 3,
      "h" : 1
    },
    "input" : {
      "k" : 10,
      "w" : 40,
      "n" : 1,
      "_rank" : 3,
      "h" : 1
    },
    "cast_11" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "123" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "171" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "y.1" : {
      "k" : 1,
      "w" : 10,
      "n" : 1,
      "_rank" : 3,
      "h" : 1
    },
    "167" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "cast_13" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "V" : {
      "k" : 10,
      "w" : 40,
      "n" : 1,
      "_rank" : 3,
      "h" : 1
    },
    "energy" : {
      "k" : 10,
      "w" : 40,
      "n" : 1,
      "_rank" : 3,
      "h" : 1
    },
    "124" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "193_load_constant_103" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "h" : 1
    },
    "collapsed_energy.1" : {
      "k" : 1,
      "w" : 40,
      "n" : 1,
      "_rank" : 3,
      "h" : 1
    },
    "_neg_y_182" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "transpose_0" : {
      "k" : 1,
      "w" : 40,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "reduce_prod_1" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "reduce_max_1_presqueeze" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "169" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "showToUser" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "181_promoted" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "cast_8" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "174" : {
      "k" : 1,
      "w" : 46,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "84_164_axis" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "h" : 1
    },
    "131" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "indices" : {
      "k" : 1,
      "w" : 46,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "ale_threshold" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "attention.Vm" : {
      "k" : 10,
      "w" : 40,
      "n" : 1,
      "_rank" : 3,
      "h" : 46
    },
    "squared" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "norm_presqueeze" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "delta" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "transpose_1" : {
      "k" : 1,
      "w" : 40,
      "n" : 1,
      "_rank" : 3,
      "h" : 10
    },
    "bigger" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "_inversed_x.3_y_0" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "134" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "feature_1" : {
      "k" : 1,
      "w" : 46,
      "n" : 1,
      "_rank" : 2,
      "h" : 40
    },
    "182" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "200_load_constant_108" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "h" : 1
    },
    "_neg_y_t_x.3" : {
      "k" : 1,
      "w" : 10,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "bigger.1" : {
      "k" : 1,
      "w" : 2,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "183" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "reduce_prod_1_presqueeze" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "_neg_y_t_x" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "norm" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "136" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "range_1d_0" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "shape_0" : {
      "k" : 1,
      "w" : 2,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "189_load_constant_102" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "h" : 1
    },
    "cast_9" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "norm.1" : {
      "k" : 1,
      "w" : 10,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "transpose_2" : {
      "k" : 10,
      "w" : 1,
      "n" : 1,
      "_rank" : 3,
      "h" : 1
    },
    "topMatchIsConfident" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "138" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "reduce_max_3" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "_inversed_x.3" : {
      "k" : 10,
      "w" : 1,
      "n" : 1,
      "_rank" : 3,
      "h" : 1
    },
    "range_1d_1" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "collapsed_energy" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "reshape_0" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 2,
      "h" : 46
    },
    "reduce_prod_0_presqueeze" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "reduce_prod_0" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "baxter_scores" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "187" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "30" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "reduce_max_0" : {
      "k" : 1,
      "w" : 10,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "27" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "v" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "y" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "cast_10" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "scatter_0" : {
      "k" : 1,
      "w" : 46,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "reduce_max_0_presqueeze" : {
      "k" : 1,
      "w" : 10,
      "n" : 1,
      "_rank" : 3,
      "h" : 1
    },
    "193" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "cast_12" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "attention.Qm" : {
      "k" : 10,
      "w" : 40,
      "n" : 1,
      "_rank" : 3,
      "h" : 46
    },
    "189" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "scatter_1" : {
      "k" : 1,
      "w" : 46,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "reshape_3" : {
      "k" : 1,
      "w" : 46,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "171_load_constant_90" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "h" : 1
    },
    "transpose_3" : {
      "k" : 1,
      "w" : 10,
      "n" : 1,
      "_rank" : 3,
      "h" : 1
    },
    "41" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "attention_combo" : {
      "k" : 1,
      "w" : 10,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "cast_6" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "38" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "147" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "reshape_4" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 2,
      "h" : 46
    },
    "t_x.3" : {
      "k" : 1,
      "w" : 10,
      "n" : 1,
      "_rank" : 3,
      "h" : 1
    },
    "104" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "embedding" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 2,
      "h" : 40
    },
    "68_promoted" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "reshape_5" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 2,
      "h" : 46
    },
    "105" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "t_x" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "collapsed_energy.5" : {
      "k" : 1,
      "w" : 40,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "slice_by_index_0" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "198" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "reduce_max_3_presqueeze" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "transpose_4" : {
      "k" : 10,
      "w" : 1,
      "n" : 1,
      "_rank" : 3,
      "h" : 40
    },
    "norm.1_presqueeze" : {
      "k" : 1,
      "w" : 10,
      "n" : 1,
      "_rank" : 3,
      "h" : 1
    },
    "baxter_confident" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "reduce_max_1" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "epi_threshold" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "reshape_8" : {
      "k" : 1,
      "w" : 46,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "inputs" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "200" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "56" : {
      "k" : 10,
      "w" : 1,
      "n" : 1,
      "_rank" : 3,
      "h" : 1
    },
    "slice_by_index_1" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "161" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "reshape_9" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 2,
      "h" : 46
    },
    "scores" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "cast_7" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "187_presqueeze" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "all_combinations" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "shape_1" : {
      "k" : 1,
      "w" : 2,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "158" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "show" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "collapsed_energy.3" : {
      "k" : 1,
      "w" : 40,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "unsorted_show" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "x.5" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "transpose_5" : {
      "k" : 1,
      "w" : 46,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "attention.Km" : {
      "k" : 10,
      "w" : 40,
      "n" : 1,
      "_rank" : 3,
      "h" : 46
    },
    "K" : {
      "k" : 10,
      "w" : 40,
      "n" : 1,
      "_rank" : 3,
      "h" : 1
    },
    "164" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "model_inputs" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 2,
      "h" : 46
    },
    "ood" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    }
  },
  "global_shapes" : {

  }
}