add inline _arg_max

This commit is contained in:
envolution 2024-11-30 14:57:57 -05:00
parent e3eac03d40
commit f1ed315249

View File

@ -526,3 +526,5 @@ cdef class precompute_hiddens:
return d_best.reshape((d_best.shape + (1,)))
return state_vector, backprop_relu
cdef inline int _arg_max(const float* scores, const int n_classes) nogil: