Searched defs:vec_host_softmax_backward_lastdim (Results 1 – 1 of 1) sorted by relevance
1148 struct vec_host_softmax_backward_lastdim { struct1150 apply(const Tensor& grad_input, const Tensor& grad, const Tensor& output) { in apply()