{ "layer_shapes" : { "output" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "input.11" : { "k" : 1, "w" : 128, "n" : 1, "_rank" : 2, "h" : 1 }, "input" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 2, "h" : 1 }, "input.5" : { "k" : 1, "w" : 256, "n" : 1, "_rank" : 2, "h" : 1 }, "linear_3" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 2, "h" : 1 }, "33" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 2, "h" : 1 }, "input.17" : { "k" : 1, "w" : 64, "n" : 1, "_rank" : 2, "h" : 1 } } }