ArmNN
 24.02
GpuFsaSoftmax.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 <armnn/Descriptors.hpp>
9 
10 #include <gpuFsa/GpuFsaBackend.hpp>
11 
12 namespace armnn
13 {
14 
15 arm_compute::Status GpuFsaSoftmaxValidate(const TensorInfo& input,
16  const TensorInfo& output,
17  const SoftmaxDescriptor& descriptor);
18 
19 void GpuFsaSoftmaxCreateOp(GpuFsaPreCompiledBlob* blob,
20  const TensorInfo& input,
21  const TensorInfo& output,
22  const SoftmaxDescriptor& descriptor);
23 
24 }
armnn::GpuFsaSoftmaxCreateOp
void GpuFsaSoftmaxCreateOp(GpuFsaPreCompiledBlob *blob, const TensorInfo &input, const TensorInfo &output, const SoftmaxDescriptor &descriptor)
Definition: GpuFsaSoftmax.cpp:63
Descriptors.hpp
armnn::Status
Status
Definition: Types.hpp:42
GpuFsaBackend.hpp
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6
armnn::GpuFsaSoftmaxValidate
arm_compute::Status GpuFsaSoftmaxValidate(const TensorInfo &input, const TensorInfo &output, const SoftmaxDescriptor &descriptor)
Definition: GpuFsaSoftmax.cpp:22