[luci-interpreter] Support int64 dtype in the CircleTranspose (Samsun…

…g#13336) This commit supports int64 dtype in the CircleTranspose. ONE-DCO-1.0-Signed-off-by: seongwoo <[email protected]>
lemmaa · Jul 2, 2024 · 5c7d0f5 · 5c7d0f5
1 parent 7d2e519
commit 5c7d0f5
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/compiler/luci-interpreter/src/kernels/Transpose.cpp b/compiler/luci-interpreter/src/kernels/Transpose.cpp
@@ -70,6 +70,11 @@ void Transpose::execute() const
                                        getTensorData<float>(input()), getTensorShape(output()),
                                        getTensorData<float>(output()));
       break;
+    case DataType::S64:
+      tflite::reference_ops::Transpose(params, getTensorShape(input()),
+                                       getTensorData<int64_t>(input()), getTensorShape(output()),
+                                       getTensorData<int64_t>(output()));
+      break;
     case DataType::U8:
       tflite::reference_ops::Transpose(params, getTensorShape(input()),
                                        getTensorData<uint8_t>(input()), getTensorShape(output()),