Searched refs:validate_arguments_logits_softmax (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/ |
| H A D | CpuSoftmaxKernel.cpp | 207 Status validate_arguments_logits_softmax(const ITensorInfo &src, const ITensorInfo &max, in validate_arguments_logits_softmax() function
255 …ARM_COMPUTE_ERROR_THROW_ON(validate_arguments_logits_softmax(*src, *max, *dst, beta, *tmp, IS_LOG)… in configure()
288 …ARM_COMPUTE_RETURN_ON_ERROR(validate_arguments_logits_softmax(*src, *max, *dst, beta, *tmp, IS_LOG… in validate()
|