Searched refs:neon_softmax_logits_1d_float (Results 1 – 4 of 4) sorted by relevance
/external/ComputeLibrary/src/cpu/kernels/softmax/generic/neon/ |
D | fp32.cpp | 34 return neon_softmax_logits_1d_float<float>(in, max, tmp, out, beta, is_log, window); in neon_fp32_softmax()
|
D | fp16.cpp | 35 return neon_softmax_logits_1d_float<float16_t>(in, max, tmp, out, beta, is_log, window); in neon_fp16_softmax()
|
D | impl.h | 41 void neon_softmax_logits_1d_float(const ITensor *in, const ITensor *max, void *const tmp,
|
D | impl.cpp | 268 void neon_softmax_logits_1d_float(const ITensor *in, const ITensor *max, void *const tmp, in neon_softmax_logits_1d_float() function 393 template void neon_softmax_logits_1d_float<float16_t>(const ITensor *in, const ITensor *max, void *… 396 template void neon_softmax_logits_1d_float<float>(const ITensor *in, const ITensor *max, void *cons…
|