blob: dd79bd3beef0490b502964a103c34b2bb9bf2c03 [file] [log] [blame]
Sadik Armagan62483be2020-10-23 17:14:43 +01001//
2// Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
3// SPDX-License-Identifier: MIT
4//
5
6#pragma once
7
Finn Williams6f9f9902020-11-13 13:23:15 +00008#include <armnn/utility/IgnoreUnused.hpp>
9
Sadik Armagan62483be2020-10-23 17:14:43 +010010#include <tensorflow/lite/builtin_ops.h>
11#include <tensorflow/lite/c/builtin_op_data.h>
12#include <tensorflow/lite/c/common.h>
13#include <tensorflow/lite/minimal_logging.h>
James Wardf89964e2020-11-09 11:57:47 +000014#include <tensorflow/lite/kernels/internal/tensor_ctypes.h>
Sadik Armagan62483be2020-10-23 17:14:43 +010015
16namespace armnnDelegate
17{
18
19TfLiteStatus VisitTransposeOperator(DelegateData& delegateData,
20 TfLiteContext* tfLiteContext,
21 TfLiteNode* tfLiteNode,
22 int nodeIndex,
James Wardf89964e2020-11-09 11:57:47 +000023 int32_t tfliteTransposeOperatorCode)
Sadik Armagan62483be2020-10-23 17:14:43 +010024{
James Wardf89964e2020-11-09 11:57:47 +000025 TF_LITE_ENSURE_STATUS(ValidateNumInputs(tfLiteContext, tfLiteNode, 2, nodeIndex));
26 TF_LITE_ENSURE_STATUS(ValidateNumOutputs(tfLiteContext, tfLiteNode, 1, nodeIndex));
Sadik Armagan62483be2020-10-23 17:14:43 +010027
James Wardf89964e2020-11-09 11:57:47 +000028 const TfLiteTensor *tfLiteTensors = tfLiteContext->tensors;
29 const TfLiteTensor& tfLiteInputTensor0 = tfLiteTensors[tfLiteNode->inputs->data[0]];
30 if (IsDynamicTensor(tfLiteInputTensor0))
31 {
32 TF_LITE_MAYBE_KERNEL_LOG(tfLiteContext,
33 "TfLiteArmnnDelegate: Dynamic input tensors are not supported in "
34 "operator #%d node #%d: ",
35 tfliteTransposeOperatorCode, nodeIndex);
Finn Williams6f9f9902020-11-13 13:23:15 +000036
James Wardf89964e2020-11-09 11:57:47 +000037 return kTfLiteError;
38 }
39
40 const TfLiteTensor& tfLiteInputTensor1 = tfLiteTensors[tfLiteNode->inputs->data[1]];
41 if (IsDynamicTensor(tfLiteInputTensor1))
42 {
43 TF_LITE_MAYBE_KERNEL_LOG(tfLiteContext,
44 "TfLiteArmnnDelegate: Dynamic input tensors are not supported in "
45 "operator #%d node #%d: ",
46 tfliteTransposeOperatorCode, nodeIndex);
47 return kTfLiteError;
48 }
49
50 const TfLiteTensor& tfLiteOutputTensor = tfLiteTensors[tfLiteNode->outputs->data[0]];
51 if (IsDynamicTensor(tfLiteOutputTensor))
52 {
53 TF_LITE_MAYBE_KERNEL_LOG(tfLiteContext,
54 "TfLiteArmnnDelegate: Dynamic output tensors are not supported in "
55 "operator #%d node #%d: ",
56 tfliteTransposeOperatorCode, nodeIndex);
57 return kTfLiteError;
58 }
59
60 const armnn::TensorInfo& inputTensorInfo0 = GetTensorInfoForTfLiteTensor(tfLiteInputTensor0);
61 const armnn::TensorInfo& inputTensorInfo1 = GetTensorInfoForTfLiteTensor(tfLiteInputTensor1); //permutation tensor
62 const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor);
63
64 auto* permTensorDataPtr = tflite::GetTensorData<int32_t>(&tfLiteInputTensor1);
James Ward7c0e3fd2020-11-16 18:46:12 +000065 unsigned int numEl = tfLiteInputTensor1.dims->data[0];
James Wardf89964e2020-11-09 11:57:47 +000066
67 ARMNN_ASSERT( numEl <= armnn::MaxNumOfTensorDimensions);
68 ARMNN_ASSERT( tfLiteInputTensor1.dims->size == 1); // ensure only single dimension to the permutation tensor
69
70 armnn::TransposeDescriptor descriptor(armnn::PermutationVector(
71 reinterpret_cast<const armnn::PermutationVector::ValueType *> (permTensorDataPtr),
Finn Williams6f9f9902020-11-13 13:23:15 +000072 static_cast<armnn::PermutationVector::SizeType>(numEl)));
James Wardf89964e2020-11-09 11:57:47 +000073
74 bool isSupported = false;
75
76 auto validateFunc = [&](const armnn::TensorInfo& outputTensorInfo, bool& isSupported)
77 {
78 FORWARD_LAYER_SUPPORT_FUNC(__func__,
79 tfLiteContext,
80 IsTransposeSupported,
81 delegateData.m_Backends,
82 isSupported,
83 inputTensorInfo0,
84 outputTensorInfo,
85 descriptor);
86 };
87
88 if (!delegateData.m_Network)
89 {
90 validateFunc(outputTensorInfo, isSupported);
91 return isSupported ? kTfLiteOk : kTfLiteError;
92 }
93
94 armnn::IConnectableLayer* transposeLayer = delegateData.m_Network->AddTransposeLayer(descriptor);
95 ARMNN_ASSERT(transposeLayer != nullptr);
96 ARMNN_ASSERT(transposeLayer->GetNumInputSlots() == 1); // permutation vector given to descriptor object
97
98 armnn::IOutputSlot& outputSlot = transposeLayer->GetOutputSlot(0);
99 outputSlot.SetTensorInfo(outputTensorInfo);
100
101 return Connect(transposeLayer, tfLiteNode, delegateData);
102}
Sadik Armagan62483be2020-10-23 17:14:43 +0100103} // namespace armnnDelegate