Home
last modified time | relevance | path

Searched defs:CreateSoftmax (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
H A Dsoftmax.cc74 GPUOperation CreateSoftmax(const OperationDef& definition) { in CreateSoftmax() function
/aosp_15_r20/external/armnn/src/backends/backendsCommon/
H A DWorkloadFactoryBase.hpp239 std::unique_ptr<IWorkload> CreateSoftmax(const SoftmaxQueueDescriptor& /*descriptor*/, in CreateSoftmax() function in armnn::WorkloadFactoryBase
H A DWorkloadFactory.cpp2280 std::unique_ptr<IWorkload> IWorkloadFactory::CreateSoftmax(const SoftmaxQueueDescriptor& /*descript… in CreateSoftmax() function in armnn::IWorkloadFactory
/aosp_15_r20/external/armnn/src/backends/reference/
H A DRefWorkloadFactory.cpp1107 std::unique_ptr<IWorkload> RefWorkloadFactory::CreateSoftmax(const SoftmaxQueueDescriptor& descript… in CreateSoftmax() function in armnn::RefWorkloadFactory
/aosp_15_r20/external/armnn/src/backends/neon/
H A DNeonWorkloadFactory.cpp1034 std::unique_ptr<IWorkload> NeonWorkloadFactory::CreateSoftmax(const SoftmaxQueueDescriptor& descrip… in CreateSoftmax() function in armnn::NeonWorkloadFactory
/aosp_15_r20/external/armnn/src/backends/cl/
H A DClWorkloadFactory.cpp1189 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateSoftmax(const SoftmaxQueueDescriptor& descripto… in CreateSoftmax() function in armnn::ClWorkloadFactory