未验证 提交 752f29a1 编写于 作者: I idontkonwher 提交者: GitHub

bug fix of operator "interp_linear"

上级 db4ae6e6
...@@ -90,7 +90,7 @@ __global__ void KeLinearInterpFw(const T* in, ...@@ -90,7 +90,7 @@ __global__ void KeLinearInterpFw(const T* in,
if (data_layout == DataLayout::kNCHW) { if (data_layout == DataLayout::kNCHW) {
const T* in_pos = const T* in_pos =
&in[out_id_h * out_id_w + channel_id * in_img_size + in_img_idx]; &in[out_id_h * input_w + channel_id * in_img_size + in_img_idx];
// linear interpolation // linear interpolation
out[out_id_h * output_w + out_id_w] = out[out_id_h * output_w + out_id_w] =
static_cast<T>(w2lambda * static_cast<MT>(in_pos[0]) + static_cast<T>(w2lambda * static_cast<MT>(in_pos[0]) +
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册