-
Notifications
You must be signed in to change notification settings - Fork 4
Closed
Milestone
Description
E.g. for Transformer test case:
extern_data_data_dim_tags_1_time_dim = SpatialDim('time')
extern_data_data_dim_tags_2_input_dim = FeatureDim('input', 10)
network_encoder_subnetwork_layers_0_subnetwork_self_attn_subnetwork_self_attn_subnetwork_qkv_split_dims_dims_0_num_heads_dim = SpatialDim('num_heads', 2)
network_encoder_subnetwork_layers_0_subnetwork_self_attn_subnetwork_self_attn_subnetwork_k_new_dim_set_dim_tags_time_encoder_layers_0_self_attn_self_attn_self_att_expand_dim_dim = SpatialDim('encoder/layers.0/self_attn/self_attn:self_att_expand_dim')
network_encoder_subnetwork_layers_1_subnetwork_self_attn_subnetwork_self_attn_subnetwork_k_new_dim_set_dim_tags_time_encoder_layers_1_self_attn_self_attn_self_att_expand_dim_dim = SpatialDim('encoder/layers.1/self_attn/self_attn:self_att_expand_dim')
network_loop_unit_state_decoder_0_self_attn_k_accum_state_out_spatial_dim_loop_decoder_layers_0_self_attn_self_attn_self_att_expand_dim_dim = SpatialDim('loop/decoder/layers.0/self_attn/self_attn:self_att_expand_dim')
network_loop_unit_state_decoder_1_self_attn_k_accum_state_out_spatial_dim_loop_decoder_layers_1_self_attn_self_attn_self_att_expand_dim_dim = SpatialDim('loop/decoder/layers.1/self_attn/self_attn:self_att_expand_dim')
network_loop_unit_target_embedding_subnetwork_dot_reduce_target_dim = FeatureDim('target', 7)
network_loop_unit_decoder_subnetwork_layers_0_subnetwork_self_attn_subnetwork_self_attn_subnetwork_qkv_split_dims_dims_0_num_heads_dim = SpatialDim('num_heads', 2)
network_loop_axis_loop_dim_dim = SpatialDim('loop-dim')
This could be way shorter.
Metadata
Metadata
Assignees
Labels
No labels