Searched refs:multivariate_normal_log_prob_gradcheck (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributions/ |
| H A D | test_distributions.py | 2897 def multivariate_normal_log_prob_gradcheck( function
2921 multivariate_normal_log_prob_gradcheck(mean, cov)
2922 multivariate_normal_log_prob_gradcheck(mean_multi_batch, cov)
2923 multivariate_normal_log_prob_gradcheck(mean_multi_batch, cov_batched)
2924 multivariate_normal_log_prob_gradcheck(mean, None, prec)
2925 multivariate_normal_log_prob_gradcheck(mean_no_batch, None, prec_batched)
2926 multivariate_normal_log_prob_gradcheck(mean, None, None, scale_tril)
2927 multivariate_normal_log_prob_gradcheck(
|