Searched refs:_log_softmax_backward_batch_rule (Results 1 – 1 of 1) sorted by relevance
289 static std::tuple<Tensor, std::optional<int64_t>> _log_softmax_backward_batch_rule( in _log_softmax_backward_batch_rule() function506 VMAP_SUPPORT(_log_softmax_backward_data, _log_softmax_backward_batch_rule); in TORCH_LIBRARY_IMPL()