Searched refs:vec_softmax (Results 1 – 3 of 3) sorted by relevance
| /external/executorch/kernels/portable/cpu/test/ |
| D | vec_ops_test.cpp | 164 torch::executor::vec_softmax(out, X, 3); in TEST()
|
| /external/executorch/kernels/portable/cpu/ |
| D | vec_ops.h | 197 inline void vec_softmax(T* __restrict__ y, const U* __restrict__ x, int n) { in vec_softmax() function
|
| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | SoftMaxKernel.cpp | 1127 struct vec_softmax { struct
1218 [&] { vec_softmax<scalar_t, false>::apply(result, self, dim); }); in softmax_kernel_impl()
1233 [&] { vec_softmax<scalar_t, true>::apply(result, self, dim); }); in log_softmax_kernel_impl()
|