ArmNN
 24.08
ReduceOperator.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2024 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #pragma once
7 
8 #include "TosaOperatorUtils.hpp"
9 
10 TosaSerializationBasicBlock* ConvertReduceToTosaOperator(const Layer* layer,
11  const std::vector<const TensorInfo*>& inputs,
12  const std::vector<const TensorInfo*>& outputs,
13  const ReduceDescriptor* reduceDescriptor);
armnn::Layer
Definition: Layer.hpp:230
TosaOperatorUtils.hpp
ConvertReduceToTosaOperator
TosaSerializationBasicBlock * ConvertReduceToTosaOperator(const Layer *layer, const std::vector< const TensorInfo * > &inputs, const std::vector< const TensorInfo * > &outputs, const ReduceDescriptor *reduceDescriptor)
Definition: ReduceOperator.cpp:17
armnn::ReduceDescriptor
A ReduceDescriptor for the REDUCE operators.
Definition: Descriptors.hpp:1538