{
  "layer_shapes" : {
    "domain_embedding_expand_dims" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "input.1" : {
      "k" : 1,
      "w" : 53,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "cast_1_ceil" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "input.9" : {
      "k" : 1,
      "w" : 64,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "cast_1_floor" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "x" : {
      "k" : 1,
      "w" : 53,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "x_1" : {
      "k" : 1,
      "w" : 53,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "domain_tag" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "input.5" : {
      "k" : 1,
      "w" : 64,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "domain_embedding" : {
      "k" : 1,
      "w" : 53,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "cast_1" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "input" : {
      "k" : 1,
      "w" : 64,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "mask" : {
      "k" : 1,
      "w" : 53,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "linear_3" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "cast_1_cond" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "cast_1_cond_load_constant_3" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "h" : 1
    }
  }
}