update timvx transpose4D (#610)

2bc19316 · BowShotDS · GitHub · 40909634 · 2bc19316
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

src/dev/tim-vx/op/timvx_transpose.cc src/dev/tim-vx/op/timvx_transpose.cc +2 -2

未找到文件。
--- a/src/dev/tim-vx/op/timvx_transpose.cc
+++ b/src/dev/tim-vx/op/timvx_transpose.cc
@@ -52,9 +52,9 @@ bool VXEngine::AddTransposeNode(struct ir_node* ir_node)
    }
    else if (output_tensor->dim_num == 4)
    {
-        perm[0] = (uint32_t )param->tr_shape[3];
+        perm[0] = (uint32_t )param->tr_shape[1];
        perm[1] = (uint32_t )param->tr_shape[0];
-        perm[2] = (uint32_t )param->tr_shape[1];
+        perm[2] = (uint32_t )param->tr_shape[3];
        perm[3] = (uint32_t )param->tr_shape[2];
    }