{ "layers" : [ { "top" : "indices", "w" : 46, "h" : 1, "weights" : { }, "nd_rank" : 1, "type" : "load_constant", "k" : 1, "attributes" : { "basic_block_head" : 1 }, "bottom" : "", "debug_info" : "indices", "n" : 1, "name" : "indices", "constant_blob" : 1 }, { "bottom" : "inputs,indices", "validate_indices" : true, "allow_negative_indices" : true, "mode" : 0, "weights" : { }, "debug_info" : "model_inputs", "axis" : 0, "top" : "model_inputs", "type" : "gather_nd", "name" : "model_inputs", "batch_dims" : 0 }, { "begin_n" : 0, "static_sizes_k" : 0, "end_mask_h" : true, "end_w" : 22, "end_h" : 0, "begin_h" : 0, "stride_n" : 1, "end_seq" : 1, "begin_mask_seq" : false, "begin_mask_h" : false, "squeeze_mask_h" : false, "stride_k" : 1, "type" : "general_slice", "begin_mask_k" : false, "static_sizes_w" : 0, "begin_w" : 21, "end_mask_k" : false, "use_static_sizes" : false, "stride_h" : 1, "static_sizes_n" : 0, "begin_mask_n" : false, "weights" : { }, "name" : "27", "stride_seq" : 1, "end_k" : 1, "static_sizes_seq" : 0, "top" : "27", "begin_seq" : 0, "squeeze_mask_k" : false, "begin_k" : 0, "end_mask_w" : false, "end_mask_seq" : false, "end_mask_n" : false, "debug_info" : "27", "stride_w" : 1, "squeeze_mask_seq" : false, "static_sizes_h" : 0, "begin_mask_w" : false, "bottom" : "model_inputs", "end_n" : 1, "squeeze_mask_w" : true, "squeeze_mask_n" : false }, { "bottom" : "27", "alpha" : 0, "operation" : 119, "weights" : { }, "fused_relu" : 0, "debug_info" : "30", "top" : "30", "type" : "elementwise", "name" : "30", "beta" : 120 }, { "bottom" : "model_inputs", "weights" : { }, "nd_mode" : 1, "axis" : -1, "debug_info" : "shape_0", "top" : "shape_0", "type" : "get_shape", "name" : "shape_0" }, { "axis_mode" : 31, "bottom" : "shape_0", "weights" : { }, "mode" : 2, "debug_info" : "reduce_prod_0", "use_version" : 1, "top" : "reduce_prod_0_presqueeze", "type" : "reduce", "name" : "reduce_prod_0_presqueeze" }, { "size_of_axes" : 0, "bottom" : "reduce_prod_0_presqueeze", "weights" : { }, "nd_axis" : 0, "debug_info" : "reduce_prod_0", "version" : 1, "top" : "reduce_prod_0", "type" : "squeeze", "name" : "reduce_prod_0", "squeeze_all" : true }, { "end" : 1, "bottom" : "reduce_prod_0", "weights" : { }, "debug_info" : "range_1d_0", "top" : "range_1d_0", "type" : "range", "name" : "range_1d_0" }, { "name" : "reshape_0", "weights" : { }, "dst_w" : 0, "version" : 1, "dst_n" : 0, "type" : "reshape", "dst_h" : 0, "mode" : 0, "attributes" : { "basic_block_head" : 1 }, "dynamic_shape" : false, "bottom" : "range_1d_0,model_inputs", "debug_info" : "reshape_0", "nd_mode" : 1, "dst_k" : 0, "dst_seq" : 1, "top" : "reshape_0" }, { "begin_n" : 0, "static_sizes_k" : 0, "end_mask_h" : true, "end_w" : 0, "end_h" : 0, "begin_h" : 0, "stride_n" : 1, "end_seq" : 1, "attributes" : { "basic_block_head" : 1 }, "begin_mask_seq" : false, "begin_mask_h" : true, "stride_k" : 1, "squeeze_mask_h" : false, "type" : "general_slice", "begin_mask_k" : false, "static_sizes_w" : 0, "begin_w" : 21, "end_mask_k" : false, "use_static_sizes" : false, "stride_h" : 1, "static_sizes_n" : 0, "begin_mask_n" : false, "weights" : { }, "name" : "slice_by_index_0", "stride_seq" : 1, "end_k" : 1, "static_sizes_seq" : 0, "top" : "slice_by_index_0", "begin_seq" : 0, "begin_k" : 0, "squeeze_mask_k" : false, "end_mask_w" : false, "end_mask_seq" : false, "end_mask_n" : false, "debug_info" : "slice_by_index_0", "stride_w" : 1, "squeeze_mask_seq" : false, "static_sizes_h" : 0, "begin_mask_w" : false, "bottom" : "reshape_0", "end_n" : 1, "squeeze_mask_w" : true, "squeeze_mask_n" : false }, { "name" : "reshape_3", "weights" : { }, "dst_w" : -1, "version" : 1, "dst_n" : 1, "dst_nd_rank" : 1, "type" : "reshape", "dst_h" : 1, "mode" : 0, "dynamic_shape" : false, "bottom" : "model_inputs", "debug_info" : "reshape_3", "dst_seq" : 1, "dst_k" : 1, "top" : "reshape_3" }, { "bottom" : "reshape_3,slice_by_index_0,30", "validate_indices" : true, "allow_negative_indices" : true, "mode" : 0, "weights" : { }, "debug_info" : "scatter_0", "axis" : 0, "top" : "scatter_0", "type" : "scatter_nd", "name" : "scatter_0", "accumulation_mode" : 0 }, { "name" : "reshape_4", "weights" : { }, "dst_w" : 0, "version" : 1, "dst_n" : 0, "type" : "reshape", "dst_h" : 0, "mode" : 0, "dynamic_shape" : false, "bottom" : "scatter_0,model_inputs", "debug_info" : "reshape_4", "nd_mode" : 1, "dst_seq" : 1, "dst_k" : 0, "top" : "reshape_4" }, { "begin_n" : 0, "static_sizes_k" : 0, "end_mask_h" : true, "end_w" : 30, "end_h" : 0, "begin_h" : 0, "stride_n" : 1, "end_seq" : 1, "attributes" : { "basic_block_head" : 1 }, "begin_mask_seq" : false, "begin_mask_h" : false, "stride_k" : 1, "squeeze_mask_h" : false, "type" : "general_slice", "begin_mask_k" : false, "static_sizes_w" : 0, "begin_w" : 29, "end_mask_k" : false, "use_static_sizes" : false, "stride_h" : 1, "static_sizes_n" : 0, "begin_mask_n" : false, "weights" : { }, "name" : "38", "stride_seq" : 1, "end_k" : 1, "static_sizes_seq" : 0, "top" : "38", "begin_seq" : 0, "begin_k" : 0, "squeeze_mask_k" : false, "end_mask_w" : false, "end_mask_seq" : false, "end_mask_n" : false, "debug_info" : "38", "stride_w" : 1, "squeeze_mask_seq" : false, "static_sizes_h" : 0, "begin_mask_w" : false, "bottom" : "reshape_4", "end_n" : 1, "squeeze_mask_w" : true, "squeeze_mask_n" : false }, { "bottom" : "38", "alpha" : 0, "operation" : 119, "weights" : { }, "fused_relu" : 0, "debug_info" : "41", "top" : "41", "type" : "elementwise", "name" : "41", "beta" : 120 }, { "bottom" : "reshape_4", "weights" : { }, "nd_mode" : 1, "axis" : -1, "debug_info" : "shape_1", "top" : "shape_1", "type" : "get_shape", "name" : "shape_1" }, { "axis_mode" : 31, "bottom" : "shape_1", "weights" : { }, "mode" : 2, "debug_info" : "reduce_prod_1", "use_version" : 1, "top" : "reduce_prod_1_presqueeze", "type" : "reduce", "name" : "reduce_prod_1_presqueeze" }, { "size_of_axes" : 0, "bottom" : "reduce_prod_1_presqueeze", "weights" : { }, "nd_axis" : 0, "debug_info" : "reduce_prod_1", "version" : 1, "top" : "reduce_prod_1", "type" : "squeeze", "name" : "reduce_prod_1", "squeeze_all" : true }, { "end" : 1, "bottom" : "reduce_prod_1", "weights" : { }, "debug_info" : "range_1d_1", "top" : "range_1d_1", "type" : "range", "name" : "range_1d_1" }, { "name" : "reshape_5", "weights" : { }, "dst_w" : 0, "version" : 1, "dst_n" : 0, "type" : "reshape", "dst_h" : 0, "mode" : 0, "attributes" : { "basic_block_head" : 1 }, "dynamic_shape" : false, "bottom" : "range_1d_1,reshape_4", "debug_info" : "reshape_5", "nd_mode" : 1, "dst_k" : 0, "dst_seq" : 1, "top" : "reshape_5" }, { "begin_n" : 0, "static_sizes_k" : 0, "end_mask_h" : true, "end_w" : 0, "end_h" : 0, "begin_h" : 0, "stride_n" : 1, "end_seq" : 1, "attributes" : { "basic_block_head" : 1 }, "begin_mask_seq" : false, "begin_mask_h" : true, "stride_k" : 1, "squeeze_mask_h" : false, "type" : "general_slice", "begin_mask_k" : false, "static_sizes_w" : 0, "begin_w" : 29, "end_mask_k" : false, "use_static_sizes" : false, "stride_h" : 1, "static_sizes_n" : 0, "begin_mask_n" : false, "weights" : { }, "name" : "slice_by_index_1", "stride_seq" : 1, "end_k" : 1, "static_sizes_seq" : 0, "top" : "slice_by_index_1", "begin_seq" : 0, "begin_k" : 0, "squeeze_mask_k" : false, "end_mask_w" : false, "end_mask_seq" : false, "end_mask_n" : false, "debug_info" : "slice_by_index_1", "stride_w" : 1, "squeeze_mask_seq" : false, "static_sizes_h" : 0, "begin_mask_w" : false, "bottom" : "reshape_5", "end_n" : 1, "squeeze_mask_w" : true, "squeeze_mask_n" : false }, { "name" : "reshape_8", "weights" : { }, "dst_w" : -1, "version" : 1, "dst_n" : 1, "dst_nd_rank" : 1, "type" : "reshape", "dst_h" : 1, "mode" : 0, "dynamic_shape" : false, "bottom" : "reshape_4", "debug_info" : "reshape_8", "dst_seq" : 1, "dst_k" : 1, "top" : "reshape_8" }, { "bottom" : "reshape_8,slice_by_index_1,41", "validate_indices" : true, "allow_negative_indices" : true, "mode" : 0, "weights" : { }, "debug_info" : "scatter_1", "axis" : 0, "top" : "scatter_1", "type" : "scatter_nd", "name" : "scatter_1", "accumulation_mode" : 0 }, { "name" : "reshape_9", "weights" : { }, "dst_w" : 0, "version" : 1, "dst_n" : 0, "type" : "reshape", "dst_h" : 0, "mode" : 0, "dynamic_shape" : false, "bottom" : "scatter_1,reshape_4", "debug_info" : "reshape_9", "nd_mode" : 1, "dst_seq" : 1, "dst_k" : 0, "top" : "reshape_9" }, { "axis_seq" : 4, "name" : "transpose_5", "axis_n" : 3, "axis_h" : 0, "type" : "transpose", "attributes" : { "basic_block_head" : 1 }, "bottom" : "reshape_9", "axis_w" : 1, "axis_k" : 2, "debug_info" : "transpose_5", "weights" : { }, "top" : "transpose_5" }, { "top" : "attention.Qm", "w" : 40, "h" : 46, "name" : "attention.Qm", "nd_rank" : 3, "type" : "load_constant", "k" : 10, "bottom" : "", "debug_info" : "attention.Qm", "n" : 1, "weights" : { }, "constant_blob" : 3 }, { "bottom" : "transpose_5,attention.Qm", "weights" : { }, "debug_info" : "Q", "top" : "Q", "type" : "batch_matmul", "name" : "Q", "channel_mode" : false }, { "top" : "attention.Km", "w" : 40, "h" : 46, "name" : "attention.Km", "nd_rank" : 3, "type" : "load_constant", "k" : 10, "bottom" : "", "debug_info" : "attention.Km", "n" : 1, "weights" : { }, "constant_blob" : 5 }, { "bottom" : "transpose_5,attention.Km", "weights" : { }, "debug_info" : "K", "top" : "K", "type" : "batch_matmul", "name" : "K", "channel_mode" : false }, { "top" : "attention.Vm", "w" : 40, "h" : 46, "name" : "attention.Vm", "nd_rank" : 3, "type" : "load_constant", "k" : 10, "bottom" : "", "debug_info" : "attention.Vm", "n" : 1, "weights" : { }, "constant_blob" : 7 }, { "bottom" : "transpose_5,attention.Vm", "weights" : { }, "debug_info" : "V", "top" : "V", "type" : "batch_matmul", "name" : "V", "channel_mode" : false }, { "axis_h" : 0, "axis_w" : 1, "bottom" : "K", "axis_k" : 2, "axis_n" : 3, "axis_seq" : 4, "weights" : { }, "debug_info" : "transpose_4", "top" : "transpose_4", "type" : "transpose", "name" : "transpose_4" }, { "bottom" : "Q,transpose_4", "weights" : { }, "debug_info" : "56", "top" : "56", "type" : "batch_matmul", "name" : "56", "channel_mode" : false }, { "top" : "_inversed_x.3_y_0", "w" : 1, "h" : 1, "name" : "_inversed_x.3_y_0", "nd_rank" : 1, "type" : "load_constant", "k" : 1, "bottom" : "", "debug_info" : "_inversed_x.3_y_0", "n" : 1, "weights" : { }, "constant_blob" : 9 }, { "bottom" : "56,_inversed_x.3_y_0", "alpha" : 1, "operation" : 1, "weights" : { }, "fused_relu" : 0, "debug_info" : "_inversed_x.3", "top" : "_inversed_x.3", "type" : "elementwise", "name" : "_inversed_x.3", "beta" : 0 }, { "axis_h" : 1, "axis_w" : 2, "bottom" : "_inversed_x.3", "axis_k" : 0, "axis_n" : 3, "axis_seq" : 4, "weights" : { }, "debug_info" : "transpose_3", "top" : "transpose_3", "type" : "transpose", "name" : "transpose_3" }, { "axis_mode" : 4, "bottom" : "transpose_3", "weights" : { }, "mode" : 9, "nd_axis" : 0, "nd_mode" : true, "debug_info" : "reduce_max_0", "use_version" : 1, "top" : "reduce_max_0_presqueeze", "type" : "reduce", "name" : "reduce_max_0_presqueeze" }, { "size_of_axes" : 1, "bottom" : "reduce_max_0_presqueeze", "axes_0" : 0, "weights" : { }, "nd_axis" : 0, "debug_info" : "reduce_max_0", "version" : 1, "top" : "reduce_max_0", "type" : "squeeze", "name" : "reduce_max_0" }, { "alpha" : -1, "bottom" : "reduce_max_0", "weights" : { }, "mode" : 6, "debug_info" : "_neg_y_t_x.3", "top" : "_neg_y_t_x.3", "type" : "activation", "name" : "_neg_y_t_x.3", "beta" : 0 }, { "bottom" : "transpose_3,_neg_y_t_x.3", "alpha" : 1, "operation" : 0, "weights" : { }, "fused_relu" : 0, "debug_info" : "t_x.3", "top" : "t_x.3", "type" : "elementwise", "name" : "t_x.3", "beta" : 0 }, { "top" : "68_promoted", "w" : 1, "h" : 1, "name" : "68_promoted", "nd_rank" : 1, "type" : "load_constant", "k" : 1, "bottom" : "", "debug_info" : "68_promoted", "n" : 1, "weights" : { }, "constant_blob" : 11 }, { "bottom" : "68_promoted,t_x.3", "alpha" : 1, "operation" : 13, "weights" : { }, "fused_relu" : 0, "debug_info" : "69", "nd_mode" : true, "top" : "69", "type" : "elementwise", "name" : "69", "beta" : 0 }, { "axis_mode" : 4, "bottom" : "69", "weights" : { }, "mode" : 0, "nd_axis" : 0, "nd_mode" : true, "debug_info" : "norm.1", "use_version" : 1, "top" : "norm.1_presqueeze", "type" : "reduce", "name" : "norm.1_presqueeze" }, { "size_of_axes" : 1, "bottom" : "norm.1_presqueeze", "axes_0" : 0, "weights" : { }, "nd_axis" : 0, "debug_info" : "norm.1", "version" : 1, "top" : "norm.1", "type" : "squeeze", "name" : "norm.1" }, { "bottom" : "69,norm.1", "alpha" : 1, "operation" : 102, "weights" : { }, "fused_relu" : 0, "debug_info" : "y.1", "nd_mode" : true, "top" : "y.1", "type" : "elementwise", "name" : "y.1", "beta" : 0 }, { "axis_h" : 1, "axis_w" : 2, "bottom" : "y.1", "axis_k" : 0, "axis_n" : 3, "axis_seq" : 4, "weights" : { }, "debug_info" : "transpose_2", "top" : "transpose_2", "type" : "transpose", "name" : "transpose_2" }, { "bottom" : "transpose_2,V", "weights" : { }, "debug_info" : "input", "top" : "input", "type" : "batch_matmul", "name" : "input", "channel_mode" : false }, { "bottom" : "input", "weights" : { }, "mode" : 0, "debug_info" : "energy", "top" : "energy", "type" : "activation", "name" : "energy" }, { "axis_h" : 2, "axis_w" : 0, "bottom" : "energy", "axis_k" : 1, "axis_n" : 3, "axis_seq" : 4, "weights" : { }, "debug_info" : "transpose_1", "top" : "transpose_1", "type" : "transpose", "name" : "transpose_1" }, { "top" : "attention_combo", "w" : 10, "h" : 1, "name" : "attention_combo", "nd_rank" : 2, "type" : "load_constant", "k" : 1, "bottom" : "", "debug_info" : "attention_combo", "n" : 1, "weights" : { }, "constant_blob" : 13 }, { "bottom" : "attention_combo,transpose_1", "weights" : { }, "debug_info" : "collapsed_energy.1", "top" : "collapsed_energy.1", "type" : "batch_matmul", "name" : "collapsed_energy.1", "channel_mode" : false }, { "size_of_axes" : 0, "bottom" : "collapsed_energy.1", "weights" : { }, "nd_axis" : 0, "debug_info" : "collapsed_energy.3", "version" : 1, "top" : "collapsed_energy.3", "type" : "squeeze", "name" : "collapsed_energy.3", "squeeze_all" : true }, { "top" : "feature_1", "w" : 46, "h" : 40, "name" : "feature_1", "nd_rank" : 2, "type" : "load_constant", "k" : 1, "bottom" : "", "debug_info" : "feature_1", "n" : 1, "weights" : { }, "constant_blob" : 15 }, { "bottom" : "feature_1,reshape_9", "weights" : { }, "debug_info" : "embedding", "top" : "embedding", "type" : "batch_matmul", "name" : "embedding", "channel_mode" : false }, { "axis_h" : 0, "axis_w" : 1, "bottom" : "embedding", "axis_k" : 2, "axis_n" : 3, "axis_seq" : 4, "weights" : { }, "debug_info" : "transpose_0", "top" : "transpose_0", "type" : "transpose", "name" : "transpose_0" }, { "bottom" : "transpose_0,collapsed_energy.3", "alpha" : 1, "operation" : 1, "weights" : { }, "fused_relu" : 0, "debug_info" : "collapsed_energy.5", "top" : "collapsed_energy.5", "type" : "elementwise", "name" : "collapsed_energy.5", "beta" : 0 }, { "nB" : 40, "top" : "collapsed_energy", "has_biases" : 1, "weights" : { }, "nC" : 1, "blob_weights" : 19, "type" : "inner_product", "has_relu" : 0, "bottom" : "collapsed_energy.5", "blob_biases" : 17, "has_tanh" : 0, "debug_info" : "collapsed_energy", "name" : "collapsed_energy", "has_prelu" : 0 }, { "size_of_axes" : 0, "bottom" : "collapsed_energy", "weights" : { }, "nd_axis" : 0, "debug_info" : "x.5", "version" : 1, "top" : "x.5", "type" : "squeeze", "name" : "x.5", "squeeze_all" : true }, { "axis_mode" : 1, "bottom" : "x.5", "weights" : { }, "mode" : 9, "nd_axis" : 0, "nd_mode" : true, "debug_info" : "reduce_max_1", "use_version" : 1, "top" : "reduce_max_1_presqueeze", "type" : "reduce", "name" : "reduce_max_1_presqueeze" }, { "size_of_axes" : 1, "bottom" : "reduce_max_1_presqueeze", "axes_0" : 0, "weights" : { }, "nd_axis" : 0, "debug_info" : "reduce_max_1", "version" : 1, "top" : "reduce_max_1", "type" : "squeeze", "name" : "reduce_max_1" }, { "alpha" : -1, "bottom" : "reduce_max_1", "weights" : { }, "mode" : 6, "debug_info" : "_neg_y_t_x", "top" : "_neg_y_t_x", "type" : "activation", "name" : "_neg_y_t_x", "beta" : 0 }, { "bottom" : "x.5,_neg_y_t_x", "alpha" : 1, "operation" : 0, "weights" : { }, "fused_relu" : 0, "debug_info" : "t_x", "top" : "t_x", "type" : "elementwise", "name" : "t_x", "beta" : 0 }, { "top" : "104", "w" : 1, "h" : 1, "name" : "104", "nd_rank" : 1, "type" : "load_constant", "k" : 1, "bottom" : "", "debug_info" : "104", "n" : 1, "weights" : { }, "constant_blob" : 21 }, { "bottom" : "104,t_x", "alpha" : 1, "operation" : 13, "weights" : { }, "fused_relu" : 0, "debug_info" : "105", "nd_mode" : true, "top" : "105", "type" : "elementwise", "name" : "105", "beta" : 0 }, { "axis_mode" : 1, "bottom" : "105", "weights" : { }, "mode" : 0, "nd_axis" : 0, "nd_mode" : true, "debug_info" : "norm", "use_version" : 1, "top" : "norm_presqueeze", "type" : "reduce", "name" : "norm_presqueeze" }, { "size_of_axes" : 1, "bottom" : "norm_presqueeze", "axes_0" : 0, "weights" : { }, "nd_axis" : 0, "debug_info" : "norm", "version" : 1, "top" : "norm", "type" : "squeeze", "name" : "norm" }, { "bottom" : "105,norm", "alpha" : 1, "operation" : 102, "weights" : { }, "fused_relu" : 0, "debug_info" : "y", "nd_mode" : true, "top" : "y", "type" : "elementwise", "name" : "y", "beta" : 0 }, { "top" : "117", "w" : 1, "h" : 1, "name" : "117", "nd_rank" : 1, "type" : "load_constant", "k" : 1, "bottom" : "", "debug_info" : "117", "n" : 1, "weights" : { }, "constant_blob" : 23 }, { "bottom" : "y,117", "alpha" : 1, "operation" : 1, "beta" : 0, "weights" : { }, "fused_relu" : 0, "debug_info" : "x_7", "top" : "scores", "type" : "elementwise", "name" : "x_7", "attributes" : { "is_output" : 1 } }, { "top" : "epi_threshold", "w" : 1, "h" : 1, "name" : "epi_threshold", "nd_rank" : 1, "type" : "load_constant", "k" : 1, "bottom" : "", "debug_info" : "epi_threshold", "n" : 1, "weights" : { }, "constant_blob" : 25 }, { "bottom" : "epi_threshold,reduce_max_1", "alpha" : 1, "operation" : 103, "weights" : { }, "fused_relu" : 0, "debug_info" : "123", "top" : "123", "type" : "elementwise", "name" : "123", "beta" : 0 }, { "size_of_axes" : 0, "bottom" : "123", "weights" : { }, "nd_axis" : 0, "debug_info" : "124", "version" : 1, "top" : "124", "type" : "squeeze", "name" : "124", "squeeze_all" : true }, { "bottom" : "124", "weights" : { }, "mode" : 6, "debug_info" : "cast_13", "top" : "cast_13", "type" : "activation", "name" : "cast_13", "beta" : 0 }, { "bottom" : "117,cast_13", "alpha" : 1, "operation" : 1, "weights" : { }, "fused_relu" : 0, "debug_info" : "ood", "top" : "ood", "type" : "elementwise", "name" : "ood", "beta" : 0 }, { "bottom" : "scores", "weights" : { }, "ascending" : false, "axis" : 0, "debug_info" : "131", "top" : "131", "type" : "argsort", "name" : "131" }, { "bottom" : "scores,131", "validate_indices" : true, "allow_negative_indices" : true, "mode" : 2, "weights" : { }, "debug_info" : "v", "axis" : -1, "top" : "v", "type" : "gather_nd", "name" : "v", "batch_dims" : 0 }, { "bottom" : "v,v", "alpha" : 1, "operation" : 1, "weights" : { }, "fused_relu" : 0, "debug_info" : "squared", "top" : "squared", "type" : "elementwise", "name" : "squared", "beta" : 0 }, { "size_of_axes" : 1, "bottom" : "scores", "axes_0" : 1, "weights" : { }, "nd_axis" : 0, "debug_info" : "134", "top" : "134", "type" : "expand_dims", "name" : "134" }, { "size_of_axes" : 1, "bottom" : "v", "axes_0" : 0, "weights" : { }, "nd_axis" : 0, "debug_info" : "136", "top" : "136", "type" : "expand_dims", "name" : "136" }, { "bottom" : "134,136", "weights" : { }, "debug_info" : "all_combinations", "top" : "all_combinations", "type" : "batch_matmul", "name" : "all_combinations", "channel_mode" : false }, { "bottom" : "all_combinations,squared", "alpha" : 1, "operation" : 100, "weights" : { }, "fused_relu" : 0, "debug_info" : "138", "top" : "138", "type" : "elementwise", "name" : "138", "beta" : 0 }, { "top" : "147", "w" : 1, "h" : 1, "name" : "147", "nd_rank" : 1, "type" : "load_constant", "k" : 1, "bottom" : "", "debug_info" : "147", "n" : 1, "weights" : { }, "constant_blob" : 27 }, { "bottom" : "147,v", "weights" : { }, "nd_mode" : true, "axis" : 0, "debug_info" : "bigger.1", "top" : "bigger.1", "type" : "general_concat", "name" : "bigger.1" }, { "bottom" : "bigger.1", "end" : -1, "start" : 0, "weights" : { }, "axis" : 0, "debug_info" : "bigger", "top" : "bigger", "type" : "slice", "name" : "bigger" }, { "bottom" : "bigger,v", "alpha" : 1, "operation" : 33, "weights" : { }, "fused_relu" : 0, "debug_info" : "delta", "nd_mode" : true, "top" : "delta", "type" : "elementwise", "name" : "delta", "beta" : 0 }, { "top" : "ale_threshold", "w" : 1, "h" : 1, "name" : "ale_threshold", "nd_rank" : 1, "type" : "load_constant", "k" : 1, "bottom" : "", "debug_info" : "ale_threshold", "n" : 1, "weights" : { }, "constant_blob" : 29 }, { "bottom" : "ale_threshold,delta", "alpha" : 1, "operation" : 105, "weights" : { }, "fused_relu" : 0, "debug_info" : "158", "top" : "158", "type" : "elementwise", "name" : "158", "beta" : 0 }, { "bottom" : "158", "weights" : { }, "mode" : 6, "debug_info" : "cast_11", "top" : "cast_11", "type" : "activation", "name" : "cast_11", "beta" : 0 }, { "bottom" : "cast_11", "alpha" : 1, "operation" : 26, "eps" : 1.4012984643248171e-45, "weights" : { }, "fused_relu" : 0, "debug_info" : "161", "top" : "161", "type" : "elementwise", "name" : "161", "beta" : 0 }, { "bottom" : "", "k" : 1, "n" : 1, "weights" : { }, "h" : 1, "debug_info" : "164", "constant_blob" : 31, "top" : "84_164_axis", "w" : 1, "type" : "load_constant", "name" : "84_164_axis" }, { "weights" : { }, "debug_info" : "164", "top" : "164", "type" : "cumsum", "name" : "164", "bottom" : "161,84_164_axis" }, { "bottom" : "164", "alpha" : 1, "operation" : 27, "weights" : { }, "fused_relu" : 0, "debug_info" : "show", "top" : "show", "type" : "elementwise", "name" : "show", "beta" : 0 }, { "size_of_axes" : 1, "bottom" : "show", "axes_0" : 1, "weights" : { }, "nd_axis" : 0, "debug_info" : "167", "top" : "167", "type" : "expand_dims", "name" : "167" }, { "bottom" : "138", "weights" : { }, "mode" : 6, "debug_info" : "cast_12", "top" : "cast_12", "type" : "activation", "name" : "cast_12", "beta" : 0 }, { "bottom" : "cast_12,167", "weights" : { }, "debug_info" : "unsorted_show", "top" : "unsorted_show", "type" : "batch_matmul", "name" : "unsorted_show", "channel_mode" : false }, { "size_of_axes" : 0, "bottom" : "unsorted_show", "weights" : { }, "nd_axis" : 0, "debug_info" : "169", "version" : 1, "top" : "169", "type" : "squeeze", "name" : "169", "squeeze_all" : true }, { "bottom" : "", "k" : 1, "n" : 1, "weights" : { }, "h" : 1, "debug_info" : "171", "constant_blob" : 33, "top" : "171_load_constant_90", "w" : 1, "type" : "load_constant", "name" : "171_load_constant_90" }, { "bottom" : "169,171_load_constant_90", "alpha" : 1, "operation" : 105, "weights" : { }, "fused_relu" : 0, "debug_info" : "171", "top" : "171", "type" : "elementwise", "name" : "171", "beta" : 0 }, { "bottom" : "171", "weights" : { }, "mode" : 6, "debug_info" : "cast_10", "top" : "cast_10", "type" : "activation", "name" : "cast_10", "beta" : 0 }, { "bottom" : "cast_10", "alpha" : 1, "operation" : 1, "beta" : 0, "weights" : { }, "fused_relu" : 0, "debug_info" : "var_173", "top" : "showToUser", "type" : "elementwise", "name" : "var_173", "attributes" : { "is_output" : 1 } }, { "top" : "174", "w" : 46, "h" : 1, "name" : "174", "nd_rank" : 2, "type" : "load_constant", "k" : 1, "bottom" : "", "debug_info" : "174", "n" : 1, "weights" : { }, "constant_blob" : 35 }, { "bottom" : "174,model_inputs", "weights" : { }, "debug_info" : "baxter_scores", "top" : "baxter_scores", "type" : "batch_matmul", "name" : "baxter_scores", "channel_mode" : false }, { "axis_mode" : 1, "bottom" : "baxter_scores", "weights" : { }, "mode" : 9, "nd_axis" : 1, "nd_mode" : true, "debug_info" : "reduce_max_3", "use_version" : 1, "top" : "reduce_max_3_presqueeze", "type" : "reduce", "name" : "reduce_max_3_presqueeze" }, { "size_of_axes" : 1, "bottom" : "reduce_max_3_presqueeze", "axes_0" : 1, "weights" : { }, "nd_axis" : 0, "debug_info" : "reduce_max_3", "version" : 1, "top" : "reduce_max_3", "type" : "squeeze", "name" : "reduce_max_3" }, { "top" : "181_promoted", "w" : 1, "h" : 1, "name" : "181_promoted", "nd_rank" : 1, "type" : "load_constant", "k" : 1, "bottom" : "", "debug_info" : "181_promoted", "n" : 1, "weights" : { }, "constant_blob" : 37 }, { "alpha" : -1, "bottom" : "181_promoted", "weights" : { }, "mode" : 6, "debug_info" : "_neg_y_182", "top" : "_neg_y_182", "type" : "activation", "name" : "_neg_y_182", "beta" : 0 }, { "bottom" : "_neg_y_182,reduce_max_3", "alpha" : 1, "operation" : 0, "weights" : { }, "fused_relu" : 0, "debug_info" : "182", "top" : "182", "type" : "elementwise", "name" : "182", "beta" : 0 }, { "bottom" : "baxter_scores,182", "alpha" : 1, "operation" : 105, "weights" : { }, "fused_relu" : 0, "debug_info" : "183", "top" : "183", "type" : "elementwise", "name" : "183", "beta" : 0 }, { "bottom" : "183", "weights" : { }, "mode" : 6, "debug_info" : "cast_9", "top" : "cast_9", "type" : "activation", "name" : "cast_9", "beta" : 0 }, { "axis_mode" : 31, "bottom" : "cast_9", "weights" : { }, "mode" : 0, "debug_info" : "187", "use_version" : 1, "top" : "187_presqueeze", "type" : "reduce", "name" : "187_presqueeze" }, { "size_of_axes" : 0, "bottom" : "187_presqueeze", "weights" : { }, "nd_axis" : 0, "debug_info" : "187", "version" : 1, "top" : "187", "type" : "squeeze", "name" : "187", "squeeze_all" : true }, { "bottom" : "", "k" : 1, "n" : 1, "weights" : { }, "h" : 1, "debug_info" : "189", "constant_blob" : 39, "top" : "189_load_constant_102", "w" : 1, "type" : "load_constant", "name" : "189_load_constant_102" }, { "bottom" : "187,189_load_constant_102", "alpha" : 1, "operation" : 100, "weights" : { }, "fused_relu" : 0, "debug_info" : "189", "top" : "189", "type" : "elementwise", "name" : "189", "beta" : 0 }, { "bottom" : "", "k" : 1, "n" : 1, "weights" : { }, "h" : 1, "debug_info" : "193", "constant_blob" : 41, "top" : "193_load_constant_103", "w" : 1, "type" : "load_constant", "name" : "193_load_constant_103" }, { "bottom" : "reduce_max_3,193_load_constant_103", "alpha" : 1, "operation" : 105, "weights" : { }, "fused_relu" : 0, "debug_info" : "193", "top" : "193", "type" : "elementwise", "name" : "193", "beta" : 0 }, { "bottom" : "193", "weights" : { }, "mode" : 6, "debug_info" : "cast_7", "top" : "cast_7", "type" : "activation", "name" : "cast_7", "beta" : 0 }, { "bottom" : "189", "weights" : { }, "mode" : 6, "debug_info" : "cast_8", "top" : "cast_8", "type" : "activation", "name" : "cast_8", "beta" : 0 }, { "bottom" : "cast_8,cast_7", "alpha" : 1, "operation" : 1, "weights" : { }, "fused_relu" : 0, "debug_info" : "baxter_confident", "top" : "baxter_confident", "type" : "elementwise", "name" : "baxter_confident", "beta" : 0 }, { "bottom" : "ood,baxter_confident", "alpha" : 1, "operation" : 0, "weights" : { }, "fused_relu" : 0, "debug_info" : "198", "top" : "198", "type" : "elementwise", "name" : "198", "beta" : 0 }, { "bottom" : "", "k" : 1, "n" : 1, "weights" : { }, "h" : 1, "debug_info" : "200", "constant_blob" : 43, "top" : "200_load_constant_108", "w" : 1, "type" : "load_constant", "name" : "200_load_constant_108" }, { "bottom" : "198,200_load_constant_108", "alpha" : 1, "operation" : 105, "weights" : { }, "fused_relu" : 0, "debug_info" : "200", "top" : "200", "type" : "elementwise", "name" : "200", "beta" : 0 }, { "bottom" : "200", "weights" : { }, "mode" : 6, "debug_info" : "cast_6", "top" : "cast_6", "type" : "activation", "name" : "cast_6", "beta" : 0 }, { "bottom" : "cast_6", "alpha" : 1, "operation" : 1, "beta" : 0, "weights" : { }, "fused_relu" : 0, "debug_info" : "var_202", "top" : "topMatchIsConfident", "type" : "elementwise", "name" : "var_202", "attributes" : { "is_output" : 1 } } ], "storage" : "model.espresso.weights", "metadata_in_weights" : [ ], "analyses" : { "N8Espresso23analysis_debug_metadataE" : { "bundle" : "", "product" : "", "version" : "", "path" : "", "mldb_token" : "mldb-94yznbpyol", "use_case" : "", "name" : "" } }, "properties" : { }, "format_version" : 300, "shape" : { "layer_shapes" : { "117" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "69" : { "k" : 1, "w" : 10, "n" : 1, "_rank" : 3, "h" : 1 }, "Q" : { "k" : 10, "w" : 40, "n" : 1, "_rank" : 3, "h" : 1 }, "input" : { "k" : 10, "w" : 40, "n" : 1, "_rank" : 3, "h" : 1 }, "cast_11" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "123" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "171" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "y.1" : { "k" : 1, "w" : 10, "n" : 1, "_rank" : 3, "h" : 1 }, "167" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 2, "h" : 1 }, "cast_13" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "V" : { "k" : 10, "w" : 40, "n" : 1, "_rank" : 3, "h" : 1 }, "energy" : { "k" : 10, "w" : 40, "n" : 1, "_rank" : 3, "h" : 1 }, "124" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "193_load_constant_103" : { "k" : 1, "w" : 1, "n" : 1, "h" : 1 }, "collapsed_energy.1" : { "k" : 1, "w" : 40, "n" : 1, "_rank" : 3, "h" : 1 }, "_neg_y_182" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "transpose_0" : { "k" : 1, "w" : 40, "n" : 1, "_rank" : 2, "h" : 1 }, "reduce_prod_1" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "reduce_max_1_presqueeze" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "169" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "showToUser" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "181_promoted" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "cast_8" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "174" : { "k" : 1, "w" : 46, "n" : 1, "_rank" : 2, "h" : 1 }, "84_164_axis" : { "k" : 1, "w" : 1, "n" : 1, "h" : 1 }, "131" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "indices" : { "k" : 1, "w" : 46, "n" : 1, "_rank" : 1, "h" : 1 }, "ale_threshold" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "attention.Vm" : { "k" : 10, "w" : 40, "n" : 1, "_rank" : 3, "h" : 46 }, "squared" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "norm_presqueeze" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "delta" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "transpose_1" : { "k" : 1, "w" : 40, "n" : 1, "_rank" : 3, "h" : 10 }, "bigger" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "_inversed_x.3_y_0" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "134" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 2, "h" : 1 }, "feature_1" : { "k" : 1, "w" : 46, "n" : 1, "_rank" : 2, "h" : 40 }, "182" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "200_load_constant_108" : { "k" : 1, "w" : 1, "n" : 1, "h" : 1 }, "_neg_y_t_x.3" : { "k" : 1, "w" : 10, "n" : 1, "_rank" : 2, "h" : 1 }, "bigger.1" : { "k" : 1, "w" : 2, "n" : 1, "_rank" : 1, "h" : 1 }, "183" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 2, "h" : 1 }, "reduce_prod_1_presqueeze" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "_neg_y_t_x" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "norm" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "136" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 2, "h" : 1 }, "range_1d_0" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "shape_0" : { "k" : 1, "w" : 2, "n" : 1, "_rank" : 1, "h" : 1 }, "189_load_constant_102" : { "k" : 1, "w" : 1, "n" : 1, "h" : 1 }, "cast_9" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 2, "h" : 1 }, "norm.1" : { "k" : 1, "w" : 10, "n" : 1, "_rank" : 2, "h" : 1 }, "transpose_2" : { "k" : 10, "w" : 1, "n" : 1, "_rank" : 3, "h" : 1 }, "topMatchIsConfident" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "138" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 2, "h" : 1 }, "reduce_max_3" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "_inversed_x.3" : { "k" : 10, "w" : 1, "n" : 1, "_rank" : 3, "h" : 1 }, "range_1d_1" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "collapsed_energy" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 2, "h" : 1 }, "reshape_0" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 2, "h" : 46 }, "reduce_prod_0_presqueeze" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "reduce_prod_0" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "baxter_scores" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 2, "h" : 1 }, "187" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "30" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "reduce_max_0" : { "k" : 1, "w" : 10, "n" : 1, "_rank" : 2, "h" : 1 }, "27" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "v" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "y" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "cast_10" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "scatter_0" : { "k" : 1, "w" : 46, "n" : 1, "_rank" : 1, "h" : 1 }, "reduce_max_0_presqueeze" : { "k" : 1, "w" : 10, "n" : 1, "_rank" : 3, "h" : 1 }, "193" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "cast_12" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 2, "h" : 1 }, "attention.Qm" : { "k" : 10, "w" : 40, "n" : 1, "_rank" : 3, "h" : 46 }, "189" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "scatter_1" : { "k" : 1, "w" : 46, "n" : 1, "_rank" : 1, "h" : 1 }, "reshape_3" : { "k" : 1, "w" : 46, "n" : 1, "_rank" : 1, "h" : 1 }, "171_load_constant_90" : { "k" : 1, "w" : 1, "n" : 1, "h" : 1 }, "transpose_3" : { "k" : 1, "w" : 10, "n" : 1, "_rank" : 3, "h" : 1 }, "41" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "attention_combo" : { "k" : 1, "w" : 10, "n" : 1, "_rank" : 2, "h" : 1 }, "cast_6" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "38" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "147" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "reshape_4" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 2, "h" : 46 }, "t_x.3" : { "k" : 1, "w" : 10, "n" : 1, "_rank" : 3, "h" : 1 }, "104" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "embedding" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 2, "h" : 40 }, "68_promoted" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "reshape_5" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 2, "h" : 46 }, "105" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "t_x" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "collapsed_energy.5" : { "k" : 1, "w" : 40, "n" : 1, "_rank" : 2, "h" : 1 }, "slice_by_index_0" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "198" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "reduce_max_3_presqueeze" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 2, "h" : 1 }, "transpose_4" : { "k" : 10, "w" : 1, "n" : 1, "_rank" : 3, "h" : 40 }, "norm.1_presqueeze" : { "k" : 1, "w" : 10, "n" : 1, "_rank" : 3, "h" : 1 }, "baxter_confident" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "reduce_max_1" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "epi_threshold" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "reshape_8" : { "k" : 1, "w" : 46, "n" : 1, "_rank" : 1, "h" : 1 }, "inputs" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 2, "h" : 1 }, "200" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "56" : { "k" : 10, "w" : 1, "n" : 1, "_rank" : 3, "h" : 1 }, "slice_by_index_1" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "161" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "reshape_9" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 2, "h" : 46 }, "scores" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "cast_7" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "187_presqueeze" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 2, "h" : 1 }, "all_combinations" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 2, "h" : 1 }, "shape_1" : { "k" : 1, "w" : 2, "n" : 1, "_rank" : 1, "h" : 1 }, "158" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "show" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "collapsed_energy.3" : { "k" : 1, "w" : 40, "n" : 1, "_rank" : 1, "h" : 1 }, "unsorted_show" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 2, "h" : 1 }, "x.5" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "transpose_5" : { "k" : 1, "w" : 46, "n" : 1, "_rank" : 2, "h" : 1 }, "attention.Km" : { "k" : 10, "w" : 40, "n" : 1, "_rank" : 3, "h" : 46 }, "K" : { "k" : 10, "w" : 40, "n" : 1, "_rank" : 3, "h" : 1 }, "164" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 }, "model_inputs" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 2, "h" : 46 }, "ood" : { "k" : 1, "w" : 1, "n" : 1, "_rank" : 1, "h" : 1 } }, "global_shapes" : { } }, "subnetworks" : { } }