diff --git a/compiler/luci-interpreter/src/kernels/Transpose.cpp b/compiler/luci-interpreter/src/kernels/Transpose.cpp index 6e1b4f911bd..725a9523c27 100644 --- 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(input()), getTensorShape(output()), getTensorData(output())); break; + case DataType::S64: + tflite::reference_ops::Transpose(params, getTensorShape(input()), + getTensorData(input()), getTensorShape(output()), + getTensorData(output())); + break; case DataType::U8: tflite::reference_ops::Transpose(params, getTensorShape(input()), getTensorData(input()), getTensorShape(output()),