__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 | |
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 | |
scalar_sample (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
setUp(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
tearDown(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
tensor_sample_1 (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
tensor_sample_2 (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_bernoulli_shape_scalar_params(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_bernoulli_shape_tensor_params(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_beta_shape_scalar_params(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_beta_shape_tensor_params(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_binomial_shape(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_binomial_shape_vectorized_n(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_categorical_shape(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_cauchy_shape_scalar_params(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_cauchy_shape_tensor_params(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_chi2_shape_scalar_params(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_chi2_shape_tensor_params(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_dirichlet_shape(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_entropy_shape(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_exponential_shape_scalar_param(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_exponential_shape_tensor_param(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_gamma_shape_scalar_params(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_gamma_shape_tensor_params(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_geometric_shape_scalar_params(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_geometric_shape_tensor_params(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_gumbel_shape_scalar_params(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_halfcauchy_shape_scalar_params(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_halfcauchy_shape_tensor_params(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_laplace_shape_scalar_params(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_laplace_shape_tensor_params(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_multinomial_shape(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_normal_shape_scalar_params(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_normal_shape_tensor_params(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_one_hot_categorical_shape(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_pareto_shape_scalar_params(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_studentT_shape_scalar_params(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_studentT_shape_tensor_params(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_uniform_shape_scalar_params(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_uniform_shape_tensor_params(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
test_weibull_scale_scalar_params(self) (defined in test_distributions.TestDistributionShapes) | test_distributions.TestDistributionShapes | |
wrap_with_cuda_memory_check(self, method) (defined in common_utils.TestCase) | common_utils.TestCase | |