Searched refs:softmax_backward_sparse_input_preprocessing (Results 1 – 4 of 4) sorted by relevance
15 TORCH_API std::tuple<Tensor, Tensor, Tensor, int64_t> softmax_backward_sparse_input_preprocessing(
33 std::tuple<Tensor, Tensor, Tensor, int64_t> softmax_backward_sparse_input_preprocessing( in softmax_backward_sparse_input_preprocessing() function
577 softmax_backward_sparse_input_preprocessing( in softmax_backward_sparse_cpu()597 softmax_backward_sparse_input_preprocessing( in log_softmax_backward_sparse_cpu()
629 softmax_backward_sparse_input_preprocessing( in softmax_backward_sparse_cuda()649 softmax_backward_sparse_input_preprocessing( in log_softmax_backward_sparse_cuda()