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
src
backends
tosaCommon
operatorMappings
ReduceOperator.hpp
Generated on Wed Aug 28 2024 14:31:52 for Arm NN by
1.8.17