未验证 提交 7dbd8b8c 编写于 作者: L liuruyan 提交者: GitHub

address question inputs is Tensor array not Dense Tensor in static amp mode. (#51476)

上级 97112dde
......@@ -438,6 +438,9 @@ def cast_model_to_fp16(program, amp_lists=None, use_fp16_guard=True):
"while",
"while_grad",
"cast",
"tensor_array_to_tensor",
"lod_array_length",
"write_to_array",
}
global_block = program.global_block()
keep_fp32_ops = set()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册