{
  "layer_shapes" : {
    "softmax\/out" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "_neg_y_mask0.2" : {
      "k" : 1,
      "w" : 500,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "neg_inf_masked_x.1" : {
      "k" : 1,
      "w" : 64,
      "n" : 1,
      "_rank" : 5,
      "h" : 500
    },
    "cast_4_floor" : {
      "k" : 1,
      "w" : 500,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "non_zero_0" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "input.1" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "32" : {
      "k" : 1,
      "w" : 689,
      "n" : 1,
      "_rank" : 3,
      "h" : 500
    },
    "cast_4_cond" : {
      "k" : 1,
      "w" : 500,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "model_dict.mean_stdv_norm_layer._mean" : {
      "k" : 1,
      "w" : 689,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "tile_0" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "h" : 1
    },
    "input.6" : {
      "k" : 1,
      "w" : 64,
      "n" : 1,
      "_rank" : 5,
      "h" : 500
    },
    "mask.2" : {
      "k" : 1,
      "w" : 500,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "aranged_lens.2" : {
      "k" : 1,
      "w" : 500,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "cast_4_cond_load_constant_17" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "h" : 1
    },
    "mask1.2" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "h" : 1
    },
    "expand_dims_1" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "62" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 5,
      "h" : 1
    },
    "cast_5" : {
      "k" : 1,
      "w" : 500,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "37" : {
      "k" : 1,
      "w" : 64,
      "n" : 1,
      "_rank" : 3,
      "h" : 500
    },
    "num_contacts" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "55" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "cast_4" : {
      "k" : 1,
      "w" : 500,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "logits.1" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "x.1" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 5,
      "h" : 1
    },
    "slice_by_index_0" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 0,
      "h" : 1
    },
    "39" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "features" : {
      "k" : 1,
      "w" : 689,
      "n" : 1,
      "_rank" : 3,
      "h" : 500
    },
    "66" : {
      "k" : 1,
      "w" : 64,
      "n" : 1,
      "_rank" : 3,
      "h" : 500
    },
    "_inversed_input.2_y_0" : {
      "k" : 1,
      "w" : 689,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "cast_4_ceil" : {
      "k" : 1,
      "w" : 500,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "expand_dims_2" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "59_load_constant_14" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "h" : 1
    },
    "59" : {
      "k" : 1,
      "w" : 500,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "reduce_max_0" : {
      "k" : 1,
      "w" : 64,
      "n" : 1,
      "_rank" : 5,
      "h" : 500
    },
    "shape_0" : {
      "k" : 1,
      "w" : 2,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "_inversed_input.2" : {
      "k" : 1,
      "w" : 689,
      "n" : 1,
      "_rank" : 3,
      "h" : 500
    },
    "67" : {
      "k" : 1,
      "w" : 64,
      "n" : 1,
      "_rank" : 5,
      "h" : 500
    },
    "mask0.2" : {
      "k" : 1,
      "w" : 500,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    }
  },
  "global_shapes" : {

  }
}