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", ...@@ -151,7 +151,7 @@ TEST_CASE("[cpu/operator] GlobalAveragePooling",
T0->getImpl()->setRawPtr(array0, in_nb_elems); T0->getImpl()->setRawPtr(array0, in_nb_elems);
// results // results
Tres->resize(dims_out); Tres->resize(dims_in);
Tres->getImpl()->setRawPtr(result, out_nb_elems); Tres->getImpl()->setRawPtr(result, out_nb_elems);
op->forwardDims(); op->forwardDims();
...@@ -222,7 +222,7 @@ TEST_CASE("[cpu/operator] GlobalAveragePooling", ...@@ -222,7 +222,7 @@ TEST_CASE("[cpu/operator] GlobalAveragePooling",
T0->getImpl()->setRawPtr(array0, in_nb_elems); T0->getImpl()->setRawPtr(array0, in_nb_elems);
// results // results
Tres->resize(dims_out); Tres->resize(dims_in);
Tres->getImpl()->setRawPtr(result, out_nb_elems); Tres->getImpl()->setRawPtr(result, out_nb_elems);
op->forwardDims(); op->forwardDims();
...@@ -348,7 +348,7 @@ TEST_CASE("[cpu/operator] GlobalAveragePooling", ...@@ -348,7 +348,7 @@ TEST_CASE("[cpu/operator] GlobalAveragePooling",
T0->getImpl()->setRawPtr(input, in_nb_elems); T0->getImpl()->setRawPtr(input, in_nb_elems);
// results // results
Tres->resize(out_dims); Tres->resize(in_dims);
Tres->getImpl()->setRawPtr(result, out_nb_elems); Tres->getImpl()->setRawPtr(result, out_nb_elems);
op->forwardDims(); op->forwardDims();
start = std::chrono::system_clock::now(); start = std::chrono::system_clock::now();
...@@ -535,7 +535,7 @@ TEST_CASE("[cpu/operator] GlobalAveragePooling", ...@@ -535,7 +535,7 @@ TEST_CASE("[cpu/operator] GlobalAveragePooling",
T0->getImpl()->setRawPtr(input, in_nb_elems); T0->getImpl()->setRawPtr(input, in_nb_elems);
// results // results
Tres->resize(out_dims); Tres->resize(in_dims);
Tres->getImpl()->setRawPtr(result, out_nb_elems); Tres->getImpl()->setRawPtr(result, out_nb_elems);
op->forwardDims(); op->forwardDims();
start = std::chrono::system_clock::now(); 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