Home
last modified time | relevance | path

Searched refs:neon_softmax_logits_1d_float (Results 1 – 4 of 4) sorted by relevance

/external/ComputeLibrary/src/cpu/kernels/softmax/generic/neon/
Dfp32.cpp34 return neon_softmax_logits_1d_float<float>(in, max, tmp, out, beta, is_log, window); in neon_fp32_softmax()
Dfp16.cpp35 return neon_softmax_logits_1d_float<float16_t>(in, max, tmp, out, beta, is_log, window); in neon_fp16_softmax()
Dimpl.h41 void neon_softmax_logits_1d_float(const ITensor *in, const ITensor *max, void *const tmp,
Dimpl.cpp268 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…