Skip to content

GitLab

Commit c48e921c authored by Davis King's avatar Davis King
Browse files

Minor cleanup

parent a07cc86a
Hide whitespace changes
Inline Side-by-side
Showing with 7 additions and 6 deletions
dlib/dnn/cudnn_dlibapi.cpp
View file @ c48e921c
...@@ -449,14 +449,15 @@ namespace dlib ...@@ -449,14 +449,15 @@ namespace dlib
const tensor& filters const tensor& filters
) )
{ {
output.set_size(out_num_samples, out_k, out_nr, out_nc); DLIB_ASSERT(is_same_object(output,data) == false,"");
DLIB_ASSERT(is_same_object(output,filters) == false,"");
output.set_size(out_num_samples, out_k, out_nr, out_nc);
// TODO, remove DLIB_ASSERT(output.num_samples() == data.num_samples(),out_num_samples << " " << data.num_samples());
DLIB_CASSERT(output.num_samples() == data.num_samples(),out_num_samples << " " << data.num_samples()); DLIB_ASSERT(output.k() == filters.num_samples(),"");
DLIB_CASSERT(output.k() == filters.num_samples(),""); DLIB_ASSERT(output.nr() == 1+(data.nr()-1)/stride_y,"");
DLIB_CASSERT(output.nr() == 1+(data.nr()-1)/stride_y,""); DLIB_ASSERT(output.nc() == 1+(data.nc()-1)/stride_x,"");
DLIB_CASSERT(output.nc() == 1+(data.nc()-1)/stride_x,"");
const float alpha = 1; const float alpha = 1;
const float beta = 0; const float beta = 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