| __init__(self, method_name='runTest') (defined in common_utils.TestCase) | common_utils.TestCase | |
| assertAlmostEqual(self, x, y, places=None, msg=None, delta=None, allow_inf=None) (defined in common_utils.TestCase) | common_utils.TestCase | |
| assertEqual(self, x, y, prec=None, message='', allow_inf=False) (defined in common_utils.TestCase) | common_utils.TestCase | |
| assertExpected(self, s, subname=None) (defined in common_utils.TestCase) | common_utils.TestCase | |
| assertExpectedInline(self, actual, expect, skip=0) (defined in expecttest.TestCase) | expecttest.TestCase | |
| assertExpectedRaises(self, exc_type, callable, args, kwargs) (defined in common_utils.TestCase) | common_utils.TestCase | |
| assertLeaksNoCudaTensors(self, name=None) (defined in common_utils.TestCase) | common_utils.TestCase | |
| assertNotEqual(self, x, y, prec=None, message='') (defined in common_utils.TestCase) | common_utils.TestCase | |
| assertObjectIn(self, obj, iterable) (defined in common_utils.TestCase) | common_utils.TestCase | |
| assertRaisesRegex (defined in common_utils.TestCase) | common_utils.TestCase | static |
| assertRegex (defined in common_utils.TestCase) | common_utils.TestCase | static |
| assertTensorsSlowEqual(self, x, y, prec=None, message='') (defined in common_utils.TestCase) | common_utils.TestCase | |
| assertWarns(self, callable, msg='') (defined in common_utils.TestCase) | common_utils.TestCase | |
| assertWarnsRegex(self, callable, regex, msg='') (defined in common_utils.TestCase) | common_utils.TestCase | |
| dist (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| genSparseTensor(self, size, sparse_dim, nnz, is_uncoalesced, device='cpu') (defined in common_utils.TestCase) | common_utils.TestCase | |
| longMessage (defined in expecttest.TestCase) | expecttest.TestCase | static |
| maxDiff (defined in common_utils.TestCase) | common_utils.TestCase | static |
| precision (defined in common_utils.TestCase) | common_utils.TestCase | static |
| safeCoalesce(self, t) (defined in common_utils.TestCase) | common_utils.TestCase | |
| safeToDense(self, t) (defined in common_utils.TestCase) | common_utils.TestCase | |
| setUp(self) (defined in common_utils.TestCase) | common_utils.TestCase | |
| test_argmax_relaxed_categorical(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_bernoulli(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_bernoulli_3d(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_bernoulli_enumerate_support(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_beta_log_prob(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_beta_sample(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_beta_shape(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_beta_underflow(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_beta_underflow_gpu(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_binomial(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_binomial_enumerate_support(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_binomial_extreme_vals(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_binomial_log_prob(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_binomial_log_prob_vectorized_count(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_binomial_vectorized_count(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_categorical_1d(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_categorical_2d(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_categorical_enumerate_support(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_cauchy(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_cdf_icdf_inverse(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_cdf_log_prob(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_chi2_sample(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_chi2_shape(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_dirichlet_log_prob(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_dirichlet_sample(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_dirichlet_shape(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_distribution_expand(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_distribution_subclass_expand(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_enumerate_support_type(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_exponential(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_exponential_sample(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_fishersnedecor(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_fishersnedecor_sample(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_gamma_gpu_sample(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_gamma_gpu_shape(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_gamma_sample(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_gamma_shape(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_geometric(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_geometric_log_prob_and_entropy(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_geometric_sample(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_gumbel(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_gumbel_sample(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_halfcauchy(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_halfnormal(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_halfnormal_logprob(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_halfnormal_sample(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_has_examples(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_independent_expand(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_independent_shape(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_invalid_parameter_broadcasting(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_laplace(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_laplace_sample(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_lazy_property_grad(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_logisticnormal(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_logisticnormal_logprob(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_logisticnormal_sample(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_lognormal(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_lognormal_logprob(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_lognormal_sample(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_lowrank_multivariate_normal_log_prob(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_lowrank_multivariate_normal_moments(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_lowrank_multivariate_normal_properties(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_lowrank_multivariate_normal_sample(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_lowrank_multivariate_normal_shape(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_multinomial_1d(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_multinomial_1d_log_prob(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_multinomial_2d(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_multivariate_normal_log_prob(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_multivariate_normal_moments(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_multivariate_normal_properties(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_multivariate_normal_sample(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_multivariate_normal_shape(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_negative_binomial(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_negative_binomial_log_prob(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_negative_binomial_log_prob_vectorized_count(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_normal(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_normal_sample(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_one_hot_categorical_1d(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_one_hot_categorical_2d(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_one_hot_categorical_enumerate_support(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_pareto(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_pareto_sample(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_poisson_gpu_sample(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_poisson_log_prob(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_poisson_sample(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_poisson_shape(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_relaxed_bernoulli(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_relaxed_one_hot_categorical_1d(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_relaxed_one_hot_categorical_2d(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_repr(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_rounded_relaxed_bernoulli(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_rsample_requires_grad(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_sample_detached(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_studentT(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_studentT_log_prob(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_studentT_sample(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_uniform(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| test_valid_parameter_broadcasting(self) (defined in test_distributions.TestDistributions) | test_distributions.TestDistributions | |
| wrap_with_cuda_memory_check(self, method) (defined in common_utils.TestCase) | common_utils.TestCase | |