{"layer_shapes":{"86":{"k":48,"w":50,"n":4,"_rank":4,"h":16},"transpose_2":{"k":1,"w":192,"n":4,"_rank":4,"h":25},"261_0":{"k":4,"w":128,"n":1,"_rank":3,"h":25},"257_batch_first_0":{"k":25,"w":256,"n":1,"_rank":3,"h":4},"135":{"k":96,"w":25,"n":4,"_rank":4,"h":8},"224_batch_first_2_5d_reverse":{"k":128,"w":1,"n":4,"_rank":5,"h":1},"ip_clusters":{"k":4,"w":1767,"n":1,"_rank":3,"h":25},"transpose_1":{"k":25,"w":192,"n":1,"_rank":3,"h":4},"173":{"k":192,"w":26,"n":4,"_rank":4,"h":4},"97":{"k":48,"w":50,"n":4,"_rank":4,"h":16},"257_batch_first_1_5d_foward":{"k":128,"w":1,"n":4,"_rank":5,"h":1},"output_label_prob_map_after_topk":{"k":4,"w":20,"n":1,"_rank":3,"h":25},"input.20":{"k":48,"w":25,"n":4,"_rank":4,"h":8},"224_batch_first_0_5d":{"k":256,"w":1,"n":4,"seq":25,"_rank":5,"h":1},"224_batch_first_2_5d_forward":{"k":128,"w":1,"n":4,"_rank":5,"h":1},"transpose_0":{"k":4,"w":256,"n":1,"_rank":3,"h":25},"input0.1":{"k":4,"w":128,"n":1,"_rank":3,"h":25},"output.1":{"k":192,"w":25,"n":4,"_rank":4,"h":1},"124":{"k":96,"w":25,"n":4,"_rank":4,"h":8},"261_1":{"k":4,"w":128,"n":1,"_rank":3,"h":25},"162":{"k":192,"w":26,"n":4,"_rank":4,"h":4},"70":{"k":24,"w":100,"n":4,"_rank":4,"h":32},"189":{"k":192,"w":25,"n":4,"_rank":4,"h":2},"228_0":{"k":25,"w":128,"n":1,"_rank":3,"h":4},"224_batch_first_0":{"k":25,"w":256,"n":1,"_rank":3,"h":4},"input.3_expanded":{"k":128,"w":1,"n":4,"seq":25,"_rank":5,"h":1},"topk_indices":{"k":4,"w":20,"n":1,"_rank":3,"h":25},"257_batch_first_1_5d_reverse":{"k":128,"w":1,"n":4,"_rank":5,"h":1},"transpose_1_expanded":{"k":192,"w":1,"n":4,"seq":25,"_rank":5,"h":1},"img_input":{"k":1,"w":100,"n":4,"_rank":4,"h":32},"224_batch_first_1_5d_reverse":{"k":128,"w":1,"n":4,"_rank":5,"h":1},"113":{"k":96,"w":25,"n":4,"_rank":4,"h":8},"224_batch_first_1_5d_foward":{"k":128,"w":1,"n":4,"_rank":5,"h":1},"input.48":{"k":192,"w":25,"n":4,"_rank":4,"h":2},"output0.1":{"k":4,"w":192,"n":1,"_rank":3,"h":25},"151":{"k":192,"w":26,"n":4,"_rank":4,"h":4},"257_batch_first_0_5d":{"k":256,"w":1,"n":4,"seq":25,"_rank":5,"h":1},"input.3":{"k":25,"w":128,"n":1,"_rank":3,"h":4},"257_batch_first_2_5d_reverse":{"k":128,"w":1,"n":4,"_rank":5,"h":1},"59":{"k":24,"w":100,"n":4,"_rank":4,"h":32},"input.10":{"k":24,"w":50,"n":4,"_rank":4,"h":16},"input.34":{"k":96,"w":26,"n":4,"_rank":4,"h":4},"228_1":{"k":25,"w":128,"n":1,"_rank":3,"h":4},"257_batch_first_2_5d_forward":{"k":128,"w":1,"n":4,"_rank":5,"h":1}}}