特性范围
数据类型
src数据类型 |
dst数据类型 |
|---|---|
f32 |
f32 |
bf16 |
bf16 |
数据排布
固定数据排布。
输入数据 |
维度 |
排布 |
|---|---|---|
src_layer |
3 |
{time_step, batch, slc} |
src_iter |
4 |
{layer_num, dir, batch, sic} |
weight_layer |
5 |
{layer_num, dir, slc, gates, dhc} |
weight_iter |
5 |
{layer_num, dir, sic, gates, dic} |
dst_layer |
3 |
{time_step, batch, dhc} |
dst_iter |
4 |
{layer_num, dir, batch, dic} |
参数约束
无
父主题： RNN