[layer] Bug fix for permute layer

Permute layer backward bug in the calculation of the reverse
direction of the permute is fixed.

Signed-off-by: Parichay Kapoor <pk.kapoor@samsung.com>

diff --git a/nntrainer/layers/permute_layer.cpp b/nntrainer/layers/permute_layer.cpp
index 9c53cc2..068d3dd 100644 (file)
--- a/nntrainer/layers/permute_layer.cpp
+++ b/nntrainer/layers/permute_layer.cpp
@@ -59,7 +59,7 @@ void PermuteLayer::finalize(InitLayerContext &context) {
 
     /*** @todo deprecate this */
     direction_str = buildTrasposeString(direction);
-    rdirection_str = buildTrasposeString(direction);
+    rdirection_str = buildTrasposeString(reverse_direction);
   };
 
   initiate_direction();