Searched defs:argmax_out (Results 1 – 2 of 2) sorted by relevance
23 Tensor& argmax_out( in argmax_out() function
1753 TORCH_IMPL_FUNC(argmax_out) in TORCH_IMPL_FUNC() argument