{
  "layer_shapes" : {
    "input.3" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "284" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input.23" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input.29__out_state" : {
      "k" : 1,
      "w" : 8,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input.6" : {
      "k" : 1,
      "w" : 260,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input.9" : {
      "k" : 1,
      "w" : 6,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input0.1" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 384
    },
    "input.29" : {
      "k" : 1,
      "w" : 12,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "y.10" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "463" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "target.1_has_output_shape_expanded__in_state" : {
      "k" : 4608,
      "w" : 1,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "input.57" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "x.2__out_state" : {
      "k" : 1,
      "w" : 15,
      "n" : 1,
      "_rank" : 4,
      "h" : 384
    },
    "expand_dims_4" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "359" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "stem.sep_module.3.tcn.2.norm_input" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "stem.sep_module.6.tcn.2.norm_input" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "conv_0" : {
      "k" : 384,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "expand_dims_10" : {
      "k" : 256,
      "w" : 8,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "conv_6" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "input.33__statecrop" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "x.1" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input.33__in_state" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "y.12" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "conv_0__stateconcat" : {
      "k" : 4,
      "w" : 320,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "input.55_0" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "x.2" : {
      "k" : 12,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 384
    },
    "input.13__in_state" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "stem.sep_module.7.tcn.6.norm_input" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "input.37" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input.59__out_state" : {
      "k" : 1,
      "w" : 64,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input.13__stateconcat" : {
      "k" : 1,
      "w" : 5,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "485" : {
      "k" : 12,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 384
    },
    "388" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "stem.sep_module.2.tcn.6.norm_input" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "input.51_0" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "stem.front_norm.norm_input" : {
      "k" : 1,
      "w" : 384,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "y.1" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "486" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 384
    },
    "input.25_0" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "y.2" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "stem.sep_module.0.tcn.2.norm_output" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "target.1_has_output_shape_expanded__stateconcat" : {
      "k" : 4608,
      "w" : 5,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "input.6__out_state" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "y.14" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input.2_0" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "conv_10" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "expand_dims_14" : {
      "k" : 256,
      "w" : 12,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "input.21_0" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "conv_13" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "stem.sep_module.2.tcn.2.norm_output" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "input.17" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "conv_16" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "y.4" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "conv_1" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "expand_dims_2" : {
      "k" : 384,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "stem.sep_module.4.tcn.2.norm_output" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "conv_7" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "stem.sep_module.2.tcn.2.norm_input" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "33" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 384
    },
    "y.6" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "expand_dims_22" : {
      "k" : 256,
      "w" : 36,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "stem.sep_module.6.tcn.2.norm_output" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "input.33__out_state" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input.4" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input.13__statecrop" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "y.8" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "expand_dims_18" : {
      "k" : 256,
      "w" : 20,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "input.7" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input.23__stateconcat" : {
      "k" : 1,
      "w" : 6,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "stem.sep_module.6.tcn.6.norm_input" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "stem.sep_module.1.tcn.6.norm_output" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "x.2__in_state" : {
      "k" : 1,
      "w" : 15,
      "n" : 1,
      "_rank" : 4,
      "h" : 384
    },
    "input.53" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "x.2__statecrop" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 384
    },
    "expand_dims_30" : {
      "k" : 256,
      "w" : 132,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "input.5_0" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "stem.sep_module.1.tcn.6.norm_input" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "input.19__in_state" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input.59" : {
      "k" : 1,
      "w" : 68,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "stem.sep_module.3.tcn.6.norm_output" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "input.39__out_state" : {
      "k" : 1,
      "w" : 16,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "expand_dims_26" : {
      "k" : 256,
      "w" : 68,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "conv_2" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "stem.sep_module.5.tcn.6.norm_output" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "target.1_has_output_shape_expanded" : {
      "k" : 12,
      "w" : 256,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "conv_8" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "input.69__in_state" : {
      "k" : 1,
      "w" : 128,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input.65_0" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "conv_0__out_state" : {
      "k" : 4,
      "w" : 64,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "input.33" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "stem.sep_module.7.tcn.6.norm_output" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "audio" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "input.49__in_state" : {
      "k" : 1,
      "w" : 32,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "x.2__stateconcat" : {
      "k" : 1,
      "w" : 19,
      "n" : 1,
      "_rank" : 4,
      "h" : 384
    },
    "stem.sep_module.1.tcn.2.norm_input" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "input.43__statecrop" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input.39" : {
      "k" : 1,
      "w" : 20,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input.61_0" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "expand_dims_0" : {
      "k" : 4,
      "w" : 256,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "expand_dims_34" : {
      "k" : 256,
      "w" : 260,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "input.67" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "99" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input.35_0" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "input.1_0" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "input.69__out_state" : {
      "k" : 1,
      "w" : 128,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input.31_0" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "input.33__stateconcat" : {
      "k" : 1,
      "w" : 8,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input.13__out_state" : {
      "k" : 1,
      "w" : 1,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "stem.sep_module.5.tcn.6.norm_input" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "conv_11" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "input.13" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "conv_14" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "conv_17" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "input.19" : {
      "k" : 1,
      "w" : 8,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "stem.sep_module.0.tcn.6.norm_input" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "input.47" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "conv_3" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "target.1_has_output_shape_expand_dim" : {
      "k" : 4608,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "input.43__in_state" : {
      "k" : 1,
      "w" : 8,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input.6__in_state" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "conv_9" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "expand_dims_8" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "input.23__in_state" : {
      "k" : 1,
      "w" : 2,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input.19__out_state" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "stem.sep_module.5.tcn.2.norm_input" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "expand_dims_12" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "203" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input.9__out_state" : {
      "k" : 1,
      "w" : 2,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input.8" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 384
    },
    "stem.sep_module.0.tcn.2.norm_input" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "input.27" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input.43__out_state" : {
      "k" : 1,
      "w" : 8,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input0.2" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 256
    },
    "stem.sep_module.1.tcn.2.norm_output" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "input.23__statecrop" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "expand_dims_20" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "input.43__stateconcat" : {
      "k" : 1,
      "w" : 12,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "76" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "stem.sep_module.3.tcn.2.norm_output" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "151" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "stem.sep_module.4.tcn.6.norm_input" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "232" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "conv_4" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "input.49__out_state" : {
      "k" : 1,
      "w" : 32,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input.9__in_state" : {
      "k" : 1,
      "w" : 2,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "target.1_has_output_shape_expanded__out_state" : {
      "k" : 4608,
      "w" : 1,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "expand_dims_16" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "stem.sep_module.5.tcn.2.norm_output" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "input1.2" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 4608
    },
    "411" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "target" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 12
    },
    "128" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input.71_0" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "input.63" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input.45_0" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "stem.sep_module.7.tcn.2.norm_output" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "stem.front_norm.norm_output" : {
      "k" : 1,
      "w" : 384,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "307" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "expand_dims_6" : {
      "k" : 256,
      "w" : 6,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "stem.sep_module.0.tcn.6.norm_output" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "input.69" : {
      "k" : 1,
      "w" : 132,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "180" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "expand_dims_24" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "input.41_0" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "stem.sep_module.4.tcn.2.norm_input" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "input.29__in_state" : {
      "k" : 1,
      "w" : 8,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input.15_0" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "stem.sep_module.7.tcn.2.norm_input" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "stem.sep_module.2.tcn.6.norm_output" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "conv_0__in_state" : {
      "k" : 4,
      "w" : 64,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "input.12" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "440" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "conv_12" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "conv_15" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "stem.sep_module.4.tcn.6.norm_output" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "255" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input.43" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input.11_0" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "336" : {
      "k" : 1,
      "w" : 4,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "expand_dims_32" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "input.23__out_state" : {
      "k" : 1,
      "w" : 2,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "input.49" : {
      "k" : 1,
      "w" : 36,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "conv_5" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "stem.sep_module.6.tcn.6.norm_output" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "input.59__in_state" : {
      "k" : 1,
      "w" : 64,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    },
    "stem.sep_module.3.tcn.6.norm_input" : {
      "k" : 1,
      "w" : 256,
      "n" : 1,
      "_rank" : 3,
      "h" : 4
    },
    "expand_dims_28" : {
      "k" : 256,
      "w" : 4,
      "n" : 1,
      "_rank" : 4,
      "h" : 1
    },
    "input.39__in_state" : {
      "k" : 1,
      "w" : 16,
      "n" : 1,
      "_rank" : 3,
      "h" : 256
    }
  }
}