__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 | |