"git@developer.sourcefind.cn:modelzoo/resnet50_tensorflow.git" did not exist on "47067b87edc8622aa871d44e517e1b452d348aa0"
Unverified Commit 00ff30af authored by Illia Silin's avatar Illia Silin Committed by GitHub
Browse files

fix a bug for 6-dim kernels (#555)

parent 147b7db5
...@@ -396,7 +396,7 @@ struct Tensor ...@@ -396,7 +396,7 @@ struct Tensor
} }
case 6: { case 6: {
auto f = [&](auto i0, auto i1, auto i2, auto i3, auto i4, auto i5) { auto f = [&](auto i0, auto i1, auto i2, auto i3, auto i4, auto i5) {
(*this)(i0, i1, i2, i3, i4) = g(i0, i1, i2, i3, i4, i5); (*this)(i0, i1, i2, i3, i4, i5) = g(i0, i1, i2, i3, i4, i5);
}; };
make_ParallelTensorFunctor(f, make_ParallelTensorFunctor(f,
mDesc.GetLengths()[0], mDesc.GetLengths()[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