Searched defs:vec_softmax (Results 1 – 2 of 2) sorted by relevance
197 inline void vec_softmax(T* __restrict__ y, const U* __restrict__ x, int n) { in vec_softmax() function
1127 struct vec_softmax { struct1128 static void apply(const Tensor& output, const Tensor& input, int64_t dim) { in apply()