Skip to content

Commit

Permalink
assert nothing
Browse files Browse the repository at this point in the history
if(input2.size(1) != input2.size(1)) return error;????
if(input2.size(1) != input3.size(1)) return error; right?
  • Loading branch information
zhhezhhe authored Aug 1, 2019
1 parent 019dccb commit 75d9b8f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion my_package/SeparableConv/separableconv_cuda.cc
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ int SeparableConvLayer_gpu_forward(
if(channel!=3) return error;
int batch = input1.size(0);
if(input2.size( 0) != batch) return error;
if(input2.size(1) != input2.size(1)) return error;
if(input2.size(1) != input3.size(1)) return error; //change by zhenghe, am I right?

int h = input1.size(2);
int w = input1.size(3);
Expand Down

0 comments on commit 75d9b8f

Please sign in to comment.