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

Working version

parent 5699a3e5
No related branches found
No related tags found
2 merge requests!93Release v0.3.0,!72Im2col
Pipeline #49629 failed
......@@ -52,8 +52,8 @@ void FoldImpl2D_cpu_forward_kernel(const Fold_Op<2>::Attrs &attrs, const std::ve
for (DimSize_t n = 0; n < dims[0]; ++n) {
for (DimSize_t outC = 0; outC < outChannels; ++outC) {
const auto inOffsetH = outC % kernelDims[1];
const auto inOffsetW = (outC / kernelDims[1]) % kernelDims[0];
const auto inOffsetW = outC % kernelDims[1];
const auto inOffsetH = (outC / kernelDims[1]) % kernelDims[0];
const auto inC = outC / kernelDims[0] / kernelDims[1];
for (DimSize_t outH = 0; outH < outHeight; ++outH) {
......
......@@ -173,6 +173,6 @@ TEST_CASE("[cpu/operator] Fold(forward)", "[Fold][CPU]") {
SequentialScheduler scheduler(g);
scheduler.forward();
opFold->getOutput(0)->print();
//opFold->getOutput(0)->print();
REQUIRE(*(opFold->getOutput(0)) == *myOutput);
}
\ No newline at end of file
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