{ "layer_shapes" : { "_inversed_sims" : { "k" : 1, "w" : 90, "n" : 1, "_rank" : 2, "h" : 90 }, "509" : { "k" : 1, "w" : 1536, "n" : 1, "_rank" : 3, "h" : 90 }, "input.23" : { "k" : 32, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "input.3" : { "k" : 8, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "445" : { "k" : 1, "w" : 90, "n" : 1, "_rank" : 2, "h" : 90 }, "534" : { "k" : 1, "w" : 128, "n" : 1, "_rank" : 3, "h" : 360 }, "x.21" : { "k" : 192, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "290:2" : { "k" : 64, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "input.9" : { "k" : 8, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "input.51" : { "k" : 256, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "input.68_5d_constant_in_bias" : { "k" : 512, "w" : 1, "n" : 1, "h" : 1 }, "input.29" : { "k" : 64, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "527" : { "k" : 1, "w" : 128, "n" : 1, "_rank" : 3, "h" : 360 }, "input.54" : { "k" : 256, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "608" : { "k" : 1, "w" : 90, "n" : 1, "_rank" : 2, "h" : 1 }, "536" : { "k" : 360, "w" : 1, "n" : 1, "_rank" : 3, "h" : 128 }, "439" : { "k" : 1, "w" : 90, "n" : 1, "_rank" : 2, "h" : 1 }, "x.22" : { "k" : 64, "w" : 54, "n" : 1, "_rank" : 4, "h" : 90 }, "period_detector_logits" : { "k" : 1, "w" : 90, "n" : 1, "_rank" : 3, "h" : 1 }, "within_period_x" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 3, "h" : 90 }, "254:0" : { "k" : 32, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "x.23" : { "k" : 54, "w" : 64, "n" : 1, "_rank" : 4, "h" : 90 }, "input.31" : { "k" : 64, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "input.34" : { "k" : 64, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "x.2" : { "k" : 90, "w" : 18, "n" : 1, "_rank" : 4, "h" : 3 }, "input.62" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 3, "h" : 90 }, "attn_output_weights.1" : { "k" : 360, "w" : 1, "n" : 1, "_rank" : 3, "h" : 1 }, "x.24" : { "k" : 18, "w" : 64, "n" : 1, "_rank" : 4, "h" : 90 }, "input.65" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 3, "h" : 90 }, "x.3" : { "k" : 3, "w" : 90, "n" : 1, "_rank" : 4, "h" : 18 }, "input.68" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 3, "h" : 90 }, "181:0" : { "k" : 8, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "x.4" : { "k" : 1, "w" : 90, "n" : 1, "_rank" : 3, "h" : 54 }, "254:1" : { "k" : 32, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "x.5" : { "k" : 3, "w" : 90, "n" : 1, "_rank" : 4, "h" : 18 }, "input.11" : { "k" : 16, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "dist" : { "k" : 1, "w" : 90, "n" : 1, "_rank" : 2, "h" : 90 }, "dist.1" : { "k" : 1, "w" : 90, "n" : 1, "_rank" : 2, "h" : 90 }, "x.6" : { "k" : 24, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "input.42" : { "k" : 128, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "x.26" : { "k" : 192, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "input.62_mvn" : { "k" : 512, "w" : 1, "n" : 90, "_rank" : 5, "h" : 1 }, "input.17" : { "k" : 16, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "x.7" : { "k" : 8, "w" : 54, "n" : 1, "_rank" : 4, "h" : 90 }, "181:1" : { "k" : 8, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "output.3" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 3, "h" : 90 }, "input.48" : { "k" : 128, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "254:2" : { "k" : 32, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "input.68_5d_constant_in_scale" : { "k" : 512, "w" : 1, "n" : 1, "h" : 1 }, "x.8" : { "k" : 54, "w" : 8, "n" : 1, "_rank" : 4, "h" : 90 }, "output.6" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 3, "h" : 90 }, "input.73" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 3, "h" : 90 }, "x.27" : { "k" : 64, "w" : 54, "n" : 1, "_rank" : 4, "h" : 90 }, "x.9" : { "k" : 18, "w" : 8, "n" : 1, "_rank" : 4, "h" : 90 }, "res.1" : { "k" : 16, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "510:0" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 3, "h" : 90 }, "input.1" : { "k" : 1, "w" : 90, "n" : 1, "_rank" : 3, "h" : 54 }, "x.28" : { "k" : 54, "w" : 64, "n" : 1, "_rank" : 4, "h" : 90 }, "input.4" : { "k" : 8, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "181:2" : { "k" : 8, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "input.25" : { "k" : 32, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "query" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 3, "h" : 90 }, "input.50" : { "k" : 256, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "218:0" : { "k" : 16, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "norm_a" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 2, "h" : 90 }, "input.28" : { "k" : 64, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "input.68_mvn" : { "k" : 512, "w" : 1, "n" : 90, "_rank" : 5, "h" : 1 }, "input.56" : { "k" : 1, "w" : 90, "n" : 1, "_rank" : 4, "h" : 90 }, "x.29" : { "k" : 18, "w" : 64, "n" : 1, "_rank" : 4, "h" : 90 }, "input.59" : { "k" : 360, "w" : 1, "n" : 1, "_rank" : 3, "h" : 1 }, "res.2" : { "k" : 32, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "392:0" : { "k" : 256, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "510:1" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 3, "h" : 90 }, "2" : { "k" : 1, "w" : 1536, "n" : 1, "_rank" : 1, "h" : 1 }, "4" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 1, "h" : 1 }, "res" : { "k" : 256, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "x.31" : { "k" : 384, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "periodicity" : { "k" : 1, "w" : 90, "n" : 1, "_rank" : 1, "h" : 1 }, "218:1" : { "k" : 16, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "embs" : { "k" : 90, "w" : 18, "n" : 1, "_rank" : 4, "h" : 256 }, "6" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 1, "h" : 1 }, "input.62_5d_constant_in_scale" : { "k" : 512, "w" : 1, "n" : 1, "h" : 1 }, "8" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 1, "h" : 1 }, "input.33" : { "k" : 64, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "input.62_5d_constant_in_bias" : { "k" : 512, "w" : 1, "n" : 1, "h" : 1 }, "input.36" : { "k" : 64, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "input.61" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 3, "h" : 90 }, "x.32" : { "k" : 128, "w" : 54, "n" : 1, "_rank" : 4, "h" : 90 }, "res.3" : { "k" : 64, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "392:1" : { "k" : 256, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "input.39" : { "k" : 64, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "510:2" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 3, "h" : 90 }, "input.64" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 3, "h" : 90 }, "period_length_logits" : { "k" : 1, "w" : 90, "n" : 1, "_rank" : 3, "h" : 32 }, "input.67" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 3, "h" : 90 }, "attn_output" : { "k" : 360, "w" : 128, "n" : 1, "_rank" : 3, "h" : 1 }, "x.33" : { "k" : 54, "w" : 128, "n" : 1, "_rank" : 4, "h" : 90 }, "218:2" : { "k" : 16, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "input.13" : { "k" : 16, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "x.34" : { "k" : 18, "w" : 128, "n" : 1, "_rank" : 4, "h" : 90 }, "res.4" : { "k" : 128, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "392:2" : { "k" : 256, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "input.16" : { "k" : 16, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "input.19" : { "k" : 32, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "input.44" : { "k" : 128, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "181_608_index_rank_preserved" : { "k" : 1, "w" : 90, "n" : 1, "_rank" : 3, "h" : 1 }, "output.2" : { "k" : 1, "w" : 1536, "n" : 1, "_rank" : 3, "h" : 90 }, "input.47" : { "k" : 128, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "output.5" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 3, "h" : 90 }, "a.1" : { "k" : 90, "w" : 18, "n" : 1, "_rank" : 3, "h" : 256 }, "input.62_5d" : { "k" : 512, "w" : 1, "n" : 90, "_rank" : 5, "h" : 1 }, "x.11" : { "k" : 48, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "input.75" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 3, "h" : 90 }, "output.8" : { "k" : 1, "w" : 32, "n" : 1, "_rank" : 3, "h" : 90 }, "input.78" : { "k" : 1, "w" : 32, "n" : 1, "_rank" : 3, "h" : 90 }, "356:0" : { "k" : 128, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "x.36" : { "k" : 768, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "x.12" : { "k" : 16, "w" : 54, "n" : 1, "_rank" : 4, "h" : 90 }, "input.2" : { "k" : 3, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "input.21" : { "k" : 32, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "input.5" : { "k" : 8, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "input.24" : { "k" : 32, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "q.2" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 3, "h" : 90 }, "320:0" : { "k" : 64, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "140" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 1, "h" : 1 }, "input.27" : { "k" : 64, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "input.52" : { "k" : 256, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "x.37" : { "k" : 256, "w" : 54, "n" : 1, "_rank" : 4, "h" : 90 }, "x.13" : { "k" : 54, "w" : 16, "n" : 1, "_rank" : 4, "h" : 90 }, "input.55" : { "k" : 256, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "input.68_reshape" : { "k" : 512, "w" : 1, "n" : 90, "_rank" : 5, "h" : 1 }, "input.58" : { "k" : 1, "w" : 2700, "n" : 1, "_rank" : 3, "h" : 90 }, "period" : { "k" : 1, "w" : 90, "n" : 1, "_rank" : 1, "h" : 1 }, "356:1" : { "k" : 128, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "x.38" : { "k" : 54, "w" : 256, "n" : 1, "_rank" : 4, "h" : 90 }, "142" : { "k" : 1, "w" : 32, "n" : 1, "_rank" : 1, "h" : 1 }, "x.14" : { "k" : 18, "w" : 16, "n" : 1, "_rank" : 4, "h" : 90 }, "135" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 1, "h" : 1 }, "320:1" : { "k" : 64, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "a" : { "k" : 1, "w" : 4608, "n" : 1, "_rank" : 2, "h" : 90 }, "input.62_5d_mul_out" : { "k" : 512, "w" : 1, "n" : 90, "_rank" : 5, "h" : 1 }, "x.39" : { "k" : 18, "w" : 256, "n" : 1, "_rank" : 4, "h" : 90 }, "input.32" : { "k" : 64, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "136" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 3, "h" : 90 }, "input.35" : { "k" : 64, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "input.60" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 3, "h" : 90 }, "input.38" : { "k" : 64, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "input.63" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 3, "h" : 90 }, "356:2" : { "k" : 128, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "x.16" : { "k" : 96, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "x.4_batch_norm" : { "k" : 54, "w" : 1, "n" : 1, "_rank" : 4, "h" : 90 }, "x.41" : { "k" : 256, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "138" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 1, "h" : 1 }, "430" : { "k" : 1, "w" : 90, "n" : 1, "_rank" : 2, "h" : 4608 }, "290:0" : { "k" : 64, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "k" : { "k" : 360, "w" : 128, "n" : 1, "_rank" : 3, "h" : 1 }, "poses" : { "k" : 90, "w" : 18, "n" : 1, "_rank" : 3, "h" : 3 }, "422" : { "k" : 1, "w" : 4608, "n" : 1, "_rank" : 2, "h" : 90 }, "320:2" : { "k" : 64, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "input.68_5d_mul_out" : { "k" : 512, "w" : 1, "n" : 90, "_rank" : 5, "h" : 1 }, "431" : { "k" : 1, "w" : 90, "n" : 1, "_rank" : 2, "h" : 90 }, "520" : { "k" : 1, "w" : 128, "n" : 1, "_rank" : 3, "h" : 360 }, "x.42" : { "k" : 30, "w" : 90, "n" : 1, "_rank" : 4, "h" : 90 }, "x.17" : { "k" : 32, "w" : 54, "n" : 1, "_rank" : 4, "h" : 90 }, "input.12" : { "k" : 16, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "input.68_5d" : { "k" : 512, "w" : 1, "n" : 90, "_rank" : 5, "h" : 1 }, "input.40" : { "k" : 64, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "q" : { "k" : 360, "w" : 128, "n" : 1, "_rank" : 3, "h" : 1 }, "input.15" : { "k" : 16, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "input.43" : { "k" : 128, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "433" : { "k" : 1, "w" : 90, "n" : 1, "_rank" : 2, "h" : 90 }, "x.43" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 3, "h" : 90 }, "x.18" : { "k" : 54, "w" : 32, "n" : 1, "_rank" : 4, "h" : 90 }, "output.1" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 3, "h" : 90 }, "input.46" : { "k" : 128, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "290:1" : { "k" : 64, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 }, "v" : { "k" : 360, "w" : 128, "n" : 1, "_rank" : 3, "h" : 1 }, "output.4" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 3, "h" : 90 }, "input.71" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 3, "h" : 90 }, "input.62_reshape" : { "k" : 512, "w" : 1, "n" : 90, "_rank" : 5, "h" : 1 }, "output.7" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 3, "h" : 90 }, "input.77" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 3, "h" : 90 }, "x" : { "k" : 1, "w" : 32, "n" : 1, "_rank" : 3, "h" : 90 }, "435" : { "k" : 1, "w" : 90, "n" : 1, "_rank" : 2, "h" : 90 }, "tsm" : { "k" : 1, "w" : 90, "n" : 1, "_rank" : 3, "h" : 90 }, "x.19" : { "k" : 18, "w" : 32, "n" : 1, "_rank" : 4, "h" : 90 }, "x.44" : { "k" : 1, "w" : 512, "n" : 1, "_rank" : 3, "h" : 90 }, "541" : { "k" : 1, "w" : 128, "n" : 1, "_rank" : 3, "h" : 360 }, "x.4_expanded" : { "k" : 54, "w" : 1, "n" : 1, "_rank" : 4, "h" : 90 }, "181_608_value" : { "k" : 1, "w" : 90, "n" : 1, "_rank" : 3, "h" : 1 }, "input.20" : { "k" : 32, "w" : 18, "n" : 1, "_rank" : 4, "h" : 90 } } }