Searched refs:RefSoftmaxWorkload (Results 1 – 8 of 8) sorted by relevance
20 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()
14 class RefSoftmaxWorkload : public RefBaseWorkload<SoftmaxQueueDescriptor> class
167 RefSoftmaxWorkload.cpp168 RefSoftmaxWorkload.hpp
98 workloads/RefSoftmaxWorkload.cpp \
576 return std::make_unique<RefSoftmaxWorkload>(*softmaxQueueDescriptor, info); in CreateWorkload()1110 return std::make_unique<RefSoftmaxWorkload>(descriptor, info); in CreateSoftmax()
680 RefCreateSoftmaxWorkloadTest<RefSoftmaxWorkload, armnn::DataType::Float32>();685 RefCreateSoftmaxWorkloadTest<RefSoftmaxWorkload, armnn::DataType::Float16>();690 RefCreateSoftmaxWorkloadTest<RefSoftmaxWorkload, armnn::DataType::QAsymmU8>();695 RefCreateSoftmaxWorkloadTest<RefSoftmaxWorkload, armnn::DataType::QSymmS16>();
127 CHECK_THROWS_AS(RefSoftmaxWorkload(invalidData, invalidInfo), armnn::InvalidArgumentException);
16419 FileName: ./src/backends/reference/workloads/RefSoftmaxWorkload.cpp16426 FileName: ./src/backends/reference/workloads/RefSoftmaxWorkload.hpp