未验证 提交 23f2a4ea 编写于 作者: Z Zhang Ting 提交者: GitHub

[cherry-pick] strided_slice grad add fp16 support (#47159)

* strided_slice grad add fp16 support
上级 66dccd7d
...@@ -64,6 +64,7 @@ PD_REGISTER_KERNEL(strided_slice_grad, ...@@ -64,6 +64,7 @@ PD_REGISTER_KERNEL(strided_slice_grad,
int64_t, int64_t,
float, float,
double, double,
phi::dtype::float16,
phi::dtype::complex<float>, phi::dtype::complex<float>,
phi::dtype::complex<double>) {} phi::dtype::complex<double>) {}
#endif #endif
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册