{
  "layer_shapes" : {
    "input1" : {
      "k" : 1,
      "w" : 512,
      "n" : 1,
      "_rank" : 2,
      "h" : 5
    },
    "embedding" : {
      "k" : 1,
      "w" : 512,
      "n" : 1,
      "_rank" : 2,
      "h" : 5
    }
  },
  "global_shapes" : {

  }
}