22.02
|
#include <armnnTestUtils/LayerTestResult.hpp>
#include <Half.hpp>
#include <armnn/backends/IBackendInternal.hpp>
#include <armnn/backends/WorkloadFactory.hpp>
Go to the source code of this file.
LayerTestResult<float, 2> CompareSoftmaxTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
armnn::IWorkloadFactory & | refWorkloadFactory, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
const armnn::ITensorHandleFactory & | refTensorHandleFactory, | ||
float | beta | ||
) |
Definition at line 719 of file SoftmaxTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<uint8_t, 2> CompareSoftmaxUint8Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
armnn::IWorkloadFactory & | refWorkloadFactory, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
const armnn::ITensorHandleFactory & | refTensorHandleFactory, | ||
float | beta | ||
) |
Definition at line 731 of file SoftmaxTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<float, 3> Simple3dAxisSoftmaxTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
float | beta, | ||
int | axis | ||
) |
Definition at line 346 of file SoftmaxTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<armnn::Half,3> Simple3dSoftmaxFloat16Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
float | beta | ||
) |
Definition at line 665 of file SoftmaxTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<float, 3> Simple3dSoftmaxTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
float | beta | ||
) |
Definition at line 335 of file SoftmaxTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<int16_t,3> Simple3dSoftmaxUint16Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
float | beta | ||
) |
Definition at line 696 of file SoftmaxTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<uint8_t,3> Simple3dSoftmaxUint8Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
float | beta | ||
) |
Definition at line 627 of file SoftmaxTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<float, 4> Simple4dAxisSoftmaxTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
float | beta, | ||
int | axis | ||
) |
Definition at line 452 of file SoftmaxTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<armnn::Half,4> Simple4dSoftmaxFloat16Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
float | beta | ||
) |
Definition at line 676 of file SoftmaxTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<float, 4> Simple4dSoftmaxTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
float | beta | ||
) |
Definition at line 441 of file SoftmaxTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<int16_t,4> Simple4dSoftmaxUint16Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
float | beta | ||
) |
Definition at line 707 of file SoftmaxTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<uint8_t,4> Simple4dSoftmaxUint8Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
float | beta | ||
) |
Definition at line 644 of file SoftmaxTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<float, 2> SimpleAxisSoftmaxTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
float | beta, | ||
int | axis | ||
) |
Definition at line 324 of file SoftmaxTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<armnn::Half,2> SimpleSoftmaxFloat16Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
float | beta | ||
) |
Definition at line 656 of file SoftmaxTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<float, 2> SimpleSoftmaxTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
float | beta | ||
) |
Definition at line 315 of file SoftmaxTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<int16_t,2> SimpleSoftmaxUint16Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
float | beta | ||
) |
Definition at line 687 of file SoftmaxTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<uint8_t, 2> SimpleSoftmaxUint8Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
float | beta | ||
) |
Definition at line 618 of file SoftmaxTestImpl.cpp.
Referenced by TEST_SUITE().