aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/backends/ClWorkloads/ClSoftmaxBaseWorkload.hpp
blob: 751d8ff1a4312b3cd48919ede45b6a53e308df80 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//

#pragma once

#include "backends/ClWorkloadUtils.hpp"

namespace armnn
{

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

} // namespace armnn