提交 1eec5a42 编写于 作者: P peizhilin

Merge remote-tracking branch 'upstream/develop' into windows/build

...@@ -66,9 +66,11 @@ static void CalcGridLocations(const platform::CPUDeviceContext& ctx, ...@@ -66,9 +66,11 @@ static void CalcGridLocations(const platform::CPUDeviceContext& ctx,
Tensor half_xmax; Tensor half_xmax;
Tensor half_ymax; Tensor half_ymax;
half_xmax.mutable_data<T>({n, h, w}, ctx.GetPlace()); half_xmax.mutable_data<T>({n, h, w}, ctx.GetPlace());
auto half_xmax_t = EigenTensor<T, 3>::From(half_xmax).setConstant(0.5 * x_max); auto half_xmax_t =
EigenTensor<T, 3>::From(half_xmax).setConstant(0.5 * x_max);
half_ymax.mutable_data<T>({n, h, w}, ctx.GetPlace()); half_ymax.mutable_data<T>({n, h, w}, ctx.GetPlace());
auto half_ymax_t = EigenTensor<T, 3>::From(half_ymax).setConstant(0.5 * y_max); auto half_ymax_t =
EigenTensor<T, 3>::From(half_ymax).setConstant(0.5 * y_max);
// scale grid to [0, h-1/w-1] // scale grid to [0, h-1/w-1]
auto grid_x_t = EigenTensor<T, 3>::From(grid_x); auto grid_x_t = EigenTensor<T, 3>::From(grid_x);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册