ait_model
X
Attributes Value
shape [512, 1024]
is_view_of None
is_output False
is_param False
dtype float16
gemm_rcr_bias_fast_gelu_0
Attributes Value
op_type gemm_rcr_bias_fast_gelu
gemm_rcr_bias_add_4
Attributes Value
op_type gemm_rcr_bias_add
dense1_weight
Attributes Value
shape [4096, 1024]
is_view_of None
is_output False
is_param True
dtype float16
dense1_bias
Attributes Value
shape [4096]
is_view_of None
is_output False
is_param True
dtype float16
gemm_rcr_bias_fast_gelu_0_0
Attributes Value
shape [512, 4096]
is_view_of None
is_output False
is_param False
dtype float16
dense2_weight
Attributes Value
shape [1024, 4096]
is_view_of None
is_output False
is_param True
dtype float16
dense2_bias
Attributes Value
shape [1024]
is_view_of None
is_output False
is_param True
dtype float16
elementwise_2_0
Attributes Value
shape [512, 1024]
is_view_of None
is_output False
is_param False
dtype float16
layernorm_3
Attributes Value
op_type layernorm
layernorm_weight
Attributes Value
shape [1024]
is_view_of None
is_output False
is_param True
dtype float16
layernorm_bias
Attributes Value
shape [1024]
is_view_of None
is_output False
is_param True
dtype float16
Y
Attributes Value
shape [512, 1024]
is_view_of None
is_output True
is_param False
dtype float16