ArmNN
 25.11
Loading...
Searching...
No Matches
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
9
10TosaSerializationBasicBlock* ConvertReduceToTosaOperator(const Layer* layer,
11 const std::vector<const TensorInfo*>& inputs,
12 const std::vector<const TensorInfo*>& outputs,
13 const ReduceDescriptor* reduceDescriptor);
TosaSerializationBasicBlock * ConvertReduceToTosaOperator(const Layer *layer, const std::vector< const TensorInfo * > &inputs, const std::vector< const TensorInfo * > &outputs, const ReduceDescriptor *reduceDescriptor)
A ReduceDescriptor for the REDUCE operators.