diff options
author | Marcin Junczys-Dowmunt <marcinjd@microsoft.com> | 2021-04-11 07:29:17 +0300 |
---|---|---|
committer | Marcin Junczys-Dowmunt <marcinjd@microsoft.com> | 2021-04-11 07:29:17 +0300 |
commit | ea5572237243e499fb249d6fb024956ddbd2c09e (patch) | |
tree | 280b5b797b70d17cc2d395dd70de458f8d937507 /src | |
parent | a05124176d8869962b717a3557c383406f8c76f4 (diff) | |
parent | a7c3a0b2ef0d3a9ffd952bb1eb97e680633d1515 (diff) |
Merge branch 'pmaster'
Diffstat (limited to 'src')
-rwxr-xr-x | src/tensors/cpu/tensor_operators.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/tensors/cpu/tensor_operators.cpp b/src/tensors/cpu/tensor_operators.cpp index 1191a2be..d92adba8 100755 --- a/src/tensors/cpu/tensor_operators.cpp +++ b/src/tensors/cpu/tensor_operators.cpp @@ -261,12 +261,12 @@ void TransposeFirst3In4(Tensor out, Tensor in, const std::vector<int>& vAxis) { // find the mapping between the transposed output dimensional indices (oi, oj, ok) // and original input dimensional indices (i, j, k) - int oi, oj, ok; #pragma omp parallel for for(int k = 0; k < l1; ++k) { int shift = k * l2 * l3; for(int j = 0; j < l2; ++j) { for(int i = 0; i < l3; ++i) { + int oi, oj, ok; if(vAxis[0] == 0) { if(vAxis[1] == 1) { oi = i; oj = j; ok = k; |