Skip to content
Snippets Groups Projects
Commit 41b356b3 authored by Olivier BICHLER's avatar Olivier BICHLER
Browse files

Fixed unit test following aidge_core#195 fix

parent 57cd0a77
No related branches found
No related tags found
2 merge requests!118v0.4.0,!109Add LRN operator
Pipeline #60164 failed
......@@ -151,7 +151,7 @@ TEST_CASE("[cpu/operator] GlobalAveragePooling",
T0->getImpl()->setRawPtr(array0, in_nb_elems);
// results
Tres->resize(dims_out);
Tres->resize(dims_in);
Tres->getImpl()->setRawPtr(result, out_nb_elems);
op->forwardDims();
......@@ -222,7 +222,7 @@ TEST_CASE("[cpu/operator] GlobalAveragePooling",
T0->getImpl()->setRawPtr(array0, in_nb_elems);
// results
Tres->resize(dims_out);
Tres->resize(dims_in);
Tres->getImpl()->setRawPtr(result, out_nb_elems);
op->forwardDims();
......@@ -348,7 +348,7 @@ TEST_CASE("[cpu/operator] GlobalAveragePooling",
T0->getImpl()->setRawPtr(input, in_nb_elems);
// results
Tres->resize(out_dims);
Tres->resize(in_dims);
Tres->getImpl()->setRawPtr(result, out_nb_elems);
op->forwardDims();
start = std::chrono::system_clock::now();
......@@ -535,7 +535,7 @@ TEST_CASE("[cpu/operator] GlobalAveragePooling",
T0->getImpl()->setRawPtr(input, in_nb_elems);
// results
Tres->resize(out_dims);
Tres->resize(in_dims);
Tres->getImpl()->setRawPtr(result, out_nb_elems);
op->forwardDims();
start = std::chrono::system_clock::now();
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment