Searched refs:log_softmax_wrapper (Results 1 – 1 of 1) sorted by relevance
109 void log_softmax_wrapper(const Tensor& X, int64_t dim, Tensor& out) { in log_softmax_wrapper() function153 log_softmax_wrapper<float>(self, dim, out); in opt_log_softmax_out()