{
  "layer_shapes" : {
    "att_softmax_0_softmax_output" : {
      "k" : 1,
      "w" : 5,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "att_softmax_0_output" : {
      "k" : 1,
      "w" : 5,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "output_1" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "input_1_reshaped_output" : {
      "k" : 1,
      "w" : 32,
      "n" : 1,
      "h" : 5
    },
    "input_1_expanded_output" : {
      "k" : 1,
      "w" : 160,
      "n" : 1,
      "_rank" : 5,
      "h" : 1
    },
    "dot_0_output" : {
      "k" : 1,
      "w" : 32,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "squeezed_1_output" : {
      "k" : 1,
      "w" : 32,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "att_softmax_0_softmax_reshaped_output" : {
      "k" : 1,
      "w" : 5,
      "n" : 1,
      "h" : 1
    },
    "input_1" : {
      "k" : 1,
      "w" : 160,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    },
    "attention_output" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 1,
      "h" : 1
    }
  }
}