Searched refs:sve_softmax_logits_1d_float (Results 1 – 4 of 4) sorted by relevance
| /external/ComputeLibrary/src/cpu/kernels/softmax/generic/sve/ |
| D | impl.cpp | 73 void sve_softmax_logits_1d_float(const ITensor *in, const ITensor *max, void *const tmp, in sve_softmax_logits_1d_float() function
174 template void sve_softmax_logits_1d_float<float>(const ITensor *in, const ITensor *max, void *const…
176 template void sve_softmax_logits_1d_float<float16_t>(const ITensor *in, const ITensor *max, void *c…
|
| D | impl.h | 36 void sve_softmax_logits_1d_float(const ITensor *in, const ITensor *max, void *const tmp,
|
| D | fp32.cpp | 35 return sve_softmax_logits_1d_float<float>(in, max, tmp, out, beta, is_log, window); in sve_fp32_softmax()
|
| D | fp16.cpp | 34 return sve_softmax_logits_1d_float<float16_t>(in, max, tmp, out, beta, is_log, window); in sve_fp16_softmax()
|