Home
last modified time | relevance | path

Searched refs:RefSoftmaxWorkload (Results 1 – 8 of 8) sorted by relevance

/external/armnn/src/backends/reference/workloads/
DRefSoftmaxWorkload.cpp20 void RefSoftmaxWorkload::Execute() const in Execute()
25 void RefSoftmaxWorkload::ExecuteAsync(ExecutionData& executionData) in ExecuteAsync()
31 void RefSoftmaxWorkload::Execute(std::vector<ITensorHandle*> inputs, std::vector<ITensorHandle*> ou… in Execute()
DRefSoftmaxWorkload.hpp14 class RefSoftmaxWorkload : public RefBaseWorkload<SoftmaxQueueDescriptor> class
DCMakeLists.txt167 RefSoftmaxWorkload.cpp
168 RefSoftmaxWorkload.hpp
/external/armnn/src/backends/reference/
Dbackend.mk98 workloads/RefSoftmaxWorkload.cpp \
DRefWorkloadFactory.cpp576 return std::make_unique<RefSoftmaxWorkload>(*softmaxQueueDescriptor, info); in CreateWorkload()
1110 return std::make_unique<RefSoftmaxWorkload>(descriptor, info); in CreateSoftmax()
/external/armnn/src/backends/reference/test/
DRefCreateWorkloadTests.cpp680 RefCreateSoftmaxWorkloadTest<RefSoftmaxWorkload, armnn::DataType::Float32>();
685 RefCreateSoftmaxWorkloadTest<RefSoftmaxWorkload, armnn::DataType::Float16>();
690 RefCreateSoftmaxWorkloadTest<RefSoftmaxWorkload, armnn::DataType::QAsymmU8>();
695 RefCreateSoftmaxWorkloadTest<RefSoftmaxWorkload, armnn::DataType::QSymmS16>();
/external/armnn/src/backends/backendsCommon/test/
DWorkloadDataValidation.cpp127 CHECK_THROWS_AS(RefSoftmaxWorkload(invalidData, invalidInfo), armnn::InvalidArgumentException);
/external/armnn/
DLICENSE.spdx16419 FileName: ./src/backends/reference/workloads/RefSoftmaxWorkload.cpp
16426 FileName: ./src/backends/reference/workloads/RefSoftmaxWorkload.hpp