{
  "layer_shapes" : {
    "input" : {
      "k" : 1,
      "w" : 316,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "9" : {
      "k" : 1,
      "w" : 316,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    },
    "embedding" : {
      "k" : 1,
      "w" : 512,
      "n" : 1,
      "_rank" : 2,
      "h" : 1
    }
  }
}