Searched refs:SoftmaxKernelInfo (Results 1 – 7 of 7) sorted by relevance
| /external/ComputeLibrary/src/gpu/cl/operators/ |
| D | ClSoftmax.h | 33 struct SoftmaxKernelInfo;
55 …Context &compile_context, const ITensorInfo &src, ITensorInfo &dst, const SoftmaxKernelInfo &info);
62 …static Status validate(const ITensorInfo &src, const ITensorInfo &dst, const SoftmaxKernelInfo &in…
|
| D | ClSoftmax.cpp | 55 …eContext &compile_context, const ITensorInfo &src, ITensorInfo &dst, const SoftmaxKernelInfo &info) in configure()
100 Status ClSoftmax::validate(const ITensorInfo &src, const ITensorInfo &dst, const SoftmaxKernelInfo … in validate()
|
| /external/ComputeLibrary/src/gpu/cl/kernels/ |
| D | ClSoftmaxKernel.h | 64 …sorInfo &src, ITensorInfo &max, ITensorInfo &dst, ITensorInfo &sum, const SoftmaxKernelInfo &info);
103 …, const ITensorInfo &src, const ITensorInfo &sum, ITensorInfo &dst, const SoftmaxKernelInfo &info);
110 …t ITensorInfo &src, const ITensorInfo &sum, const ITensorInfo &dst, const SoftmaxKernelInfo &info);
|
| D | ClSoftmaxKernel.cpp | 120 …st ITensorInfo &src, const ITensorInfo &sum, const ITensorInfo &dst, const SoftmaxKernelInfo &info) in validate_arguments_1DNorm()
162 …nsorInfo &src, ITensorInfo &max, ITensorInfo &dst, ITensorInfo &sum, const SoftmaxKernelInfo &info) in configure()
268 …t, const ITensorInfo &src, const ITensorInfo &sum, ITensorInfo &dst, const SoftmaxKernelInfo &info) in configure()
312 …st ITensorInfo &src, const ITensorInfo &sum, const ITensorInfo &dst, const SoftmaxKernelInfo &info) in validate()
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLSoftmaxLayer.cpp | 74 SoftmaxKernelInfo softmax_info{ beta, IS_LOG, input->info()->data_type(), axis }; in configure()
84 SoftmaxKernelInfo softmax_info{ beta, IS_LOG, input->data_type(), axis }; in validate()
|
| /external/ComputeLibrary/arm_compute/core/ |
| D | KernelDescriptors.h | 125 struct SoftmaxKernelInfo struct
|
| /external/ComputeLibrary/utils/ |
| D | TypePrinter.h | 3089 inline ::std::ostream &operator<<(::std::ostream &os, const SoftmaxKernelInfo &info)
3104 inline std::string to_string(const SoftmaxKernelInfo &info) in to_string()
|