未验证 提交 8a100774 编写于 作者: L lilong12 提交者: GitHub

add int8 for cast (#42634)

上级 6ac08db5
...@@ -156,7 +156,7 @@ REGISTER_OP_CPU_KERNEL( ...@@ -156,7 +156,7 @@ REGISTER_OP_CPU_KERNEL(
ops::CastOpKernel<CPU, double>, ops::CastOpKernel<CPU, int>, ops::CastOpKernel<CPU, double>, ops::CastOpKernel<CPU, int>,
ops::CastOpKernel<CPU, int64_t>, ops::CastOpKernel<CPU, int>, ops::CastOpKernel<CPU, int64_t>, ops::CastOpKernel<CPU, int>,
ops::CastOpKernel<CPU, int16_t>, ops::CastOpKernel<CPU, bool>, ops::CastOpKernel<CPU, int16_t>, ops::CastOpKernel<CPU, bool>,
ops::CastOpKernel<CPU, uint8_t>, ops::CastOpKernel<CPU, uint8_t>, ops::CastOpKernel<CPU, int8_t>,
ops::CastOpKernel<CPU, paddle::platform::float16>, ops::CastOpKernel<CPU, paddle::platform::float16>,
ops::CastOpKernel<CPU, paddle::platform::bfloat16>, ops::CastOpKernel<CPU, paddle::platform::bfloat16>,
ops::CastOpKernel<CPU, paddle::platform::complex<float>>, ops::CastOpKernel<CPU, paddle::platform::complex<float>>,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册