aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/backends/ClWorkloads/ClSoftmaxBaseWorkload.hpp
blob: e0113134af673a1bc9fb6e3e0fc6ad75389bba79 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
//
// Copyright © 2017 Arm Ltd. All rights reserved.
// See LICENSE file in the project root for full license information.
//

#pragma once

#include "backends/ClWorkloadUtils.hpp"

namespace armnn
{

arm_compute::Status ClSoftmaxWorkloadValidate(const TensorInfo& input,
                                              const TensorInfo& output);

} // namespace armnn