ArmNN
 24.02
TransposeOperator.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2022-2023 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #pragma once
7 
8 #include "TosaOperatorUtils.hpp"
9 
10 using namespace armnn;
11 using namespace tosa;
12 
13 TosaSerializationBasicBlock* ConvertTransposeToTosaOperator(const Layer* layer,
14  const std::vector<const TensorInfo*>& inputs,
15  const std::vector<const TensorInfo*>& outputs,
16  const TransposeDescriptor* transposeDescriptor);
ConvertTransposeToTosaOperator
TosaSerializationBasicBlock * ConvertTransposeToTosaOperator(const Layer *layer, const std::vector< const TensorInfo * > &inputs, const std::vector< const TensorInfo * > &outputs, const TransposeDescriptor *transposeDescriptor)
Definition: TransposeOperator.cpp:8
armnn::Layer
Definition: Layer.hpp:230
armnn::TransposeDescriptor
A TransposeDescriptor for the TransposeLayer.
Definition: Descriptors.hpp:1490
TosaOperatorUtils.hpp
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6