Skip to content

GitLab

Commit 115c5bfc authored by Jing Zhang's avatar Jing Zhang
Browse files

fixed

parent 49103987
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
host/host_tensor/include/host_conv_wrw.hpp
View file @ 115c5bfc
...@@ -29,7 +29,7 @@ void host_direct_convolution_backward_weights( ...@@ -29,7 +29,7 @@ void host_direct_convolution_backward_weights(
for(int ho = 0; ho < out.mDesc.GetLengths()[2]; ++ho) for(int ho = 0; ho < out.mDesc.GetLengths()[2]; ++ho)
{ {
int hi = ho * conv_strides[I0] + y * conv_dilations[I0] - in_left_pads[I0]; int hi = ho * conv_strides[I0] + y * conv_dilations[I0] - in_left_pads[I0];
for(int wo = 0; wo < wei.mDesc.GetLengths()[3]; ++wo) for(int wo = 0; wo < out.mDesc.GetLengths()[3]; ++wo)
{ {
int wi = wo * conv_strides[I1] + x * conv_dilations[I1] - in_left_pads[I1]; int wi = wo * conv_strides[I1] + x * conv_dilations[I1] - in_left_pads[I1];
if(hi >= 0 && hi < in.mDesc.GetLengths()[2] && wi >= 0 && if(hi >= 0 && hi < in.mDesc.GetLengths()[2] && wi >= 0 &&
...@@ -51,7 +51,7 @@ void host_direct_convolution_backward_weights( ...@@ -51,7 +51,7 @@ void host_direct_convolution_backward_weights(
for(int ho = 0; ho < out.mDesc.GetLengths()[1]; ++ho) for(int ho = 0; ho < out.mDesc.GetLengths()[1]; ++ho)
{ {
int hi = ho * conv_strides[I0] + y * conv_dilations[I0] - in_left_pads[I0]; int hi = ho * conv_strides[I0] + y * conv_dilations[I0] - in_left_pads[I0];
for(int wo = 0; wo < wei.mDesc.GetLengths()[2]; ++wo) for(int wo = 0; wo < out.mDesc.GetLengths()[2]; ++wo)
{ {
int wi = wo * conv_strides[I1] + x * conv_dilations[I1] - in_left_pads[I1]; int wi = wo * conv_strides[I1] + x * conv_dilations[I1] - in_left_pads[I1];
if(hi >= 0 && hi < in.mDesc.GetLengths()[1] && wi >= 0 && if(hi >= 0 && hi < in.mDesc.GetLengths()[1] && wi >= 0 &&
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment