Skip to content

P
Paddle

机器未来 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 43e8274a 编写于 作者: S superjomn
浏览文件
操作

merge upstream

上级 7963f735 016445c4
隐藏空白更改
内联 并排
Showing with 2 addition and 2 deletion
paddle/fluid/lite/kernels/x86/sgd_compute.cc
浏览文件 @ 43e8274a
...@@ -49,8 +49,8 @@ class SGDCompute : public KernelLite<TARGET(kX86), PRECISION(kFloat)> { ...@@ -49,8 +49,8 @@ class SGDCompute : public KernelLite<TARGET(kX86), PRECISION(kFloat)> {
const T *param_data = param->template data<T>(); const T *param_data = param->template data<T>();
const T *grad_data = grad->template data<T>(); const T *grad_data = grad->template data<T>();
int64_t rows_idx = 0; int64_t rows_idx = 0;
T *out_data = T *out_data = param_out->template mutable_data<T>(
param_out->mutable_data<T>(context.x86_device_context()->GetPlace()); context.x86_device_context()->GetPlace());
auto sgd = auto sgd =
paddle::operators::jit::KernelFuncs<paddle::operators::jit::SgdTuple<T>, paddle::operators::jit::KernelFuncs<paddle::operators::jit::SgdTuple<T>,
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或