blob: 6e96c2d2b4d68cca01298805f731ce014daa94d8 [file] [log] [blame]
telsoa01c577f2c2018-08-31 09:22:23 +01001//
2// Copyright © 2017 Arm Ltd. All rights reserved.
David Beckecb56cd2018-09-05 12:52:57 +01003// SPDX-License-Identifier: MIT
telsoa01c577f2c2018-08-31 09:22:23 +01004//
5
6#pragma once
7
Aron Virginas-Tarc9cc8042018-11-01 16:15:57 +00008#include <neon/workloads/NeonWorkloadUtils.hpp>
telsoa01c577f2c2018-08-31 09:22:23 +01009
10namespace armnn
11{
12
13arm_compute::Status NeonSoftmaxWorkloadValidate(const TensorInfo& input,
14 const TensorInfo& output,
15 const SoftmaxDescriptor& descriptor);
16
17} //namespace armnn