Path Lines of Code discussion/neutra/neutra_kernel_test.py 91 discussion/pathfinder/pathfinder_test.py 88 discussion/turnkey_inference_candidate/window_tune_nuts_sampling_test.py 38 spinoffs/fun_mc/fun_mc/backends/jax_integration_test.py 11 spinoffs/fun_mc/fun_mc/backends/tensorflow_integration_test.py 11 spinoffs/fun_mc/fun_mc/fun_mc_test.py 1369 spinoffs/fun_mc/fun_mc/prefab_test.py 258 spinoffs/fun_mc/fun_mc/sga_hmc_test.py 251 spinoffs/fun_mc/fun_mc/test_util.py 30 spinoffs/fun_mc/fun_mc/util_tfp_test.py 137 spinoffs/inference_gym/inference_gym/backends/jax_integration_test.py 8 spinoffs/inference_gym/inference_gym/backends/numpy_integration_test.py 8 spinoffs/inference_gym/inference_gym/backends/tensorflow_integration_test.py 10 spinoffs/inference_gym/inference_gym/internal/array_to_source_test.py 13 spinoffs/inference_gym/inference_gym/internal/data_test.py 83 spinoffs/inference_gym/inference_gym/internal/ground_truth_encoding_test.py 44 spinoffs/inference_gym/inference_gym/internal/test_util.py 396 spinoffs/inference_gym/inference_gym/internal/test_util_test.py 152 spinoffs/inference_gym/inference_gym/targets/banana_test.py 21 spinoffs/inference_gym/inference_gym/targets/bayesian_model_test.py 43 spinoffs/inference_gym/inference_gym/targets/brownian_motion_test.py 105 spinoffs/inference_gym/inference_gym/targets/eight_schools_test.py 29 spinoffs/inference_gym/inference_gym/targets/ill_conditioned_gaussian_test.py 38 spinoffs/inference_gym/inference_gym/targets/item_response_theory_test.py 96 spinoffs/inference_gym/inference_gym/targets/log_gaussian_cox_process_test.py 65 spinoffs/inference_gym/inference_gym/targets/logistic_regression_test.py 71 spinoffs/inference_gym/inference_gym/targets/lorenz_system_test.py 131 spinoffs/inference_gym/inference_gym/targets/model_test.py 66 spinoffs/inference_gym/inference_gym/targets/neals_funnel_test.py 21 spinoffs/inference_gym/inference_gym/targets/non_identifiable_quartic_test.py 17 spinoffs/inference_gym/inference_gym/targets/plasma_spectroscopy_test.py 67 spinoffs/inference_gym/inference_gym/targets/probit_regression_test.py 72 spinoffs/inference_gym/inference_gym/targets/radon_contextual_effects_test.py 171 spinoffs/inference_gym/inference_gym/targets/sparse_logistic_regression_test.py 99 spinoffs/inference_gym/inference_gym/targets/stochastic_volatility_test.py 68 spinoffs/inference_gym/inference_gym/targets/vector_model_test.py 247 spinoffs/inference_gym/inference_gym/targets/vectorized_stochastic_volatility_test.py 277 spinoffs/oryx/oryx/bijectors/bijector_extensions_test.py 46 spinoffs/oryx/oryx/core/interpreters/harvest_test.py 388 spinoffs/oryx/oryx/core/interpreters/inverse/custom_inverse_test.py 123 spinoffs/oryx/oryx/core/interpreters/inverse/inverse_test.py 305 spinoffs/oryx/oryx/core/interpreters/inverse/slice_test.py 65 spinoffs/oryx/oryx/core/interpreters/log_prob_test.py 102 spinoffs/oryx/oryx/core/interpreters/propagate_test.py 215 spinoffs/oryx/oryx/core/interpreters/unzip_test.py 267 spinoffs/oryx/oryx/core/kwargs_util_test.py 103 spinoffs/oryx/oryx/core/ppl/effect_handler_test.py 113 spinoffs/oryx/oryx/core/ppl/transformations_test.py 339 spinoffs/oryx/oryx/core/serialize_test.py 32 spinoffs/oryx/oryx/core/state/function_test.py 164 spinoffs/oryx/oryx/core/state/registrations_test.py 47 spinoffs/oryx/oryx/distributions/distribution_extensions_test.py 216 spinoffs/oryx/oryx/experimental/autoconj/addn_test.py 52 spinoffs/oryx/oryx/experimental/autoconj/canonicalize_test.py 234 spinoffs/oryx/oryx/experimental/autoconj/einsum_test.py 70 spinoffs/oryx/oryx/experimental/matching/jax_rewrite_test.py 153 spinoffs/oryx/oryx/experimental/matching/matcher_test.py 188 spinoffs/oryx/oryx/experimental/matching/rules_test.py 87 spinoffs/oryx/oryx/experimental/mcmc/kernels_test.py 77 spinoffs/oryx/oryx/experimental/nn/base_test.py 94 spinoffs/oryx/oryx/experimental/nn/combinator_test.py 167 spinoffs/oryx/oryx/experimental/nn/convolution_test.py 156 spinoffs/oryx/oryx/experimental/nn/core_test.py 197 spinoffs/oryx/oryx/experimental/nn/function_test.py 387 spinoffs/oryx/oryx/experimental/nn/normalization_test.py 181 spinoffs/oryx/oryx/experimental/nn/pooling_test.py 159 spinoffs/oryx/oryx/experimental/nn/reshape_test.py 45 spinoffs/oryx/oryx/experimental/optimizers/optix_test.py 130 spinoffs/oryx/oryx/internal/test_util.py 6 spinoffs/oryx/oryx/util/summary_test.py 36 tensorflow_probability/python/bijectors/absolute_value_test.py 39 tensorflow_probability/python/bijectors/ascending_test.py 69 tensorflow_probability/python/bijectors/batch_normalization_test.py 200 tensorflow_probability/python/bijectors/bijector_composition_test.py 55 tensorflow_probability/python/bijectors/bijector_properties_test.py 660 tensorflow_probability/python/bijectors/bijector_test.py 936 tensorflow_probability/python/bijectors/blockwise_test.py 281 tensorflow_probability/python/bijectors/categorical_to_discrete_test.py 96 tensorflow_probability/python/bijectors/chain_test.py 356 tensorflow_probability/python/bijectors/cholesky_outer_product_test.py 119 tensorflow_probability/python/bijectors/cholesky_to_inv_cholesky_test.py 92 tensorflow_probability/python/bijectors/correlation_cholesky_test.py 235 tensorflow_probability/python/bijectors/cumsum_test.py 67 tensorflow_probability/python/bijectors/discrete_cosine_transform_test.py 56 tensorflow_probability/python/bijectors/exp_test.py 69 tensorflow_probability/python/bijectors/expm1_test.py 61 tensorflow_probability/python/bijectors/ffjord_test.py 156 tensorflow_probability/python/bijectors/fill_scale_tril_test.py 30 tensorflow_probability/python/bijectors/fill_triangular_test.py 59 tensorflow_probability/python/bijectors/frechet_cdf_test.py 67 tensorflow_probability/python/bijectors/generalized_pareto_test.py 73 tensorflow_probability/python/bijectors/gev_cdf_test.py 55 tensorflow_probability/python/bijectors/glow_test.py 297 tensorflow_probability/python/bijectors/gompertz_cdf_test.py 59 tensorflow_probability/python/bijectors/gumbel_cdf_test.py 48 tensorflow_probability/python/bijectors/householder_test.py 46 tensorflow_probability/python/bijectors/identity_test.py 32 tensorflow_probability/python/bijectors/inline_test.py 64 tensorflow_probability/python/bijectors/invert_test.py 116 tensorflow_probability/python/bijectors/iterated_sigmoid_centered_test.py 100 tensorflow_probability/python/bijectors/joint_map_test.py 164 tensorflow_probability/python/bijectors/kumaraswamy_cdf_test.py 107 tensorflow_probability/python/bijectors/lambertw_transform_test.py 117 tensorflow_probability/python/bijectors/ldj_ratio_test.py 72 tensorflow_probability/python/bijectors/masked_autoregressive_test.py 772 tensorflow_probability/python/bijectors/matrix_inverse_tril_test.py 108 tensorflow_probability/python/bijectors/moyal_cdf_test.py 49 tensorflow_probability/python/bijectors/normal_cdf_test.py 44 tensorflow_probability/python/bijectors/ordered_test.py 70 tensorflow_probability/python/bijectors/pad_test.py 114 tensorflow_probability/python/bijectors/permute_test.py 100 tensorflow_probability/python/bijectors/power_test.py 83 tensorflow_probability/python/bijectors/power_transform_test.py 46 tensorflow_probability/python/bijectors/rational_quadratic_spline_test.py 229 tensorflow_probability/python/bijectors/rayleigh_cdf_test.py 59 tensorflow_probability/python/bijectors/real_nvp_test.py 279 tensorflow_probability/python/bijectors/reciprocal_test.py 64 tensorflow_probability/python/bijectors/reshape_test.py 327 tensorflow_probability/python/bijectors/restructure_test.py 126 tensorflow_probability/python/bijectors/scale_matvec_diag_test.py 52 tensorflow_probability/python/bijectors/scale_matvec_linear_operator_test.py 180 tensorflow_probability/python/bijectors/scale_matvec_lu_test.py 140 tensorflow_probability/python/bijectors/scale_matvec_tril_test.py 43 tensorflow_probability/python/bijectors/scale_test.py 111 tensorflow_probability/python/bijectors/shift_test.py 28 tensorflow_probability/python/bijectors/shifted_gompertz_cdf_test.py 40 tensorflow_probability/python/bijectors/sigmoid_test.py 123 tensorflow_probability/python/bijectors/sinh_arcsinh_test.py 140 tensorflow_probability/python/bijectors/sinh_test.py 40 tensorflow_probability/python/bijectors/soft_clip_test.py 91 tensorflow_probability/python/bijectors/softfloor_test.py 112 tensorflow_probability/python/bijectors/softmax_centered_test.py 117 tensorflow_probability/python/bijectors/softplus_test.py 134 tensorflow_probability/python/bijectors/softsign_test.py 61 tensorflow_probability/python/bijectors/split_test.py 274 tensorflow_probability/python/bijectors/square_test.py 33 tensorflow_probability/python/bijectors/tanh_test.py 58 tensorflow_probability/python/bijectors/transform_diagonal_test.py 90 tensorflow_probability/python/bijectors/transpose_test.py 208 tensorflow_probability/python/bijectors/weibull_cdf_test.py 78 tensorflow_probability/python/distributions/autoregressive_test.py 236 tensorflow_probability/python/distributions/batch_broadcast_test.py 220 tensorflow_probability/python/distributions/batch_concat_test.py 205 tensorflow_probability/python/distributions/batch_reshape_test.py 529 tensorflow_probability/python/distributions/bates_test.py 304 tensorflow_probability/python/distributions/bernoulli_test.py 472 tensorflow_probability/python/distributions/beta_binomial_test.py 299 tensorflow_probability/python/distributions/beta_quotient_test.py 160 tensorflow_probability/python/distributions/beta_test.py 431 tensorflow_probability/python/distributions/binomial_test.py 381 tensorflow_probability/python/distributions/blockwise_test.py 411 tensorflow_probability/python/distributions/categorical_test.py 468 tensorflow_probability/python/distributions/cauchy_test.py 370 tensorflow_probability/python/distributions/chi2_test.py 131 tensorflow_probability/python/distributions/chi_test.py 109 tensorflow_probability/python/distributions/cholesky_lkj_test.py 101 tensorflow_probability/python/distributions/cholesky_util_test.py 36 tensorflow_probability/python/distributions/continuous_bernoulli_test.py 569 tensorflow_probability/python/distributions/deterministic_test.py 407 tensorflow_probability/python/distributions/dirichlet_multinomial_test.py 423 tensorflow_probability/python/distributions/dirichlet_test.py 272 tensorflow_probability/python/distributions/discrete_rejection_sampling_test.py 34 tensorflow_probability/python/distributions/distribution_properties_test.py 434 tensorflow_probability/python/distributions/distribution_test.py 804 tensorflow_probability/python/distributions/doublesided_maxwell_test.py 88 tensorflow_probability/python/distributions/dpp_test.py 291 tensorflow_probability/python/distributions/empirical_test.py 625 tensorflow_probability/python/distributions/exp_gamma_test.py 542 tensorflow_probability/python/distributions/exponential_test.py 163 tensorflow_probability/python/distributions/exponentially_modified_gaussian_test.py 210 tensorflow_probability/python/distributions/finite_discrete_test.py 370 tensorflow_probability/python/distributions/gamma_gamma_test.py 196 tensorflow_probability/python/distributions/gamma_test.py 770 tensorflow_probability/python/distributions/gaussian_process_regression_model_test.py 446 tensorflow_probability/python/distributions/gaussian_process_test.py 258 tensorflow_probability/python/distributions/generalized_normal_test.py 410 tensorflow_probability/python/distributions/generalized_pareto_test.py 307 tensorflow_probability/python/distributions/geometric_test.py 237 tensorflow_probability/python/distributions/gev_test.py 379 tensorflow_probability/python/distributions/gumbel_test.py 238 tensorflow_probability/python/distributions/half_cauchy_test.py 453 tensorflow_probability/python/distributions/half_normal_test.py 267 tensorflow_probability/python/distributions/half_student_t_test.py 487 tensorflow_probability/python/distributions/hidden_markov_model_test.py 1728 tensorflow_probability/python/distributions/horseshoe_test.py 187 tensorflow_probability/python/distributions/independent_test.py 449 tensorflow_probability/python/distributions/internal/correlation_matrix_volumes_test.py 89 tensorflow_probability/python/distributions/internal/statistical_testing_test.py 383 tensorflow_probability/python/distributions/inverse_gamma_test.py 310 tensorflow_probability/python/distributions/inverse_gaussian_test.py 378 tensorflow_probability/python/distributions/jax_transformation_test.py 331 tensorflow_probability/python/distributions/johnson_su_test.py 419 tensorflow_probability/python/distributions/joint_distribution_auto_batched_test.py 727 tensorflow_probability/python/distributions/joint_distribution_coroutine_test.py 972 tensorflow_probability/python/distributions/joint_distribution_named_test.py 501 tensorflow_probability/python/distributions/joint_distribution_sequential_test.py 639 tensorflow_probability/python/distributions/joint_distribution_util_test.py 72 tensorflow_probability/python/distributions/kullback_leibler_test.py 118 tensorflow_probability/python/distributions/kumaraswamy_test.py 323 tensorflow_probability/python/distributions/lambertw_f_test.py 154 tensorflow_probability/python/distributions/laplace_test.py 308 tensorflow_probability/python/distributions/linear_gaussian_ssm_test.py 1318 tensorflow_probability/python/distributions/lkj_test.py 323 tensorflow_probability/python/distributions/logistic_test.py 167 tensorflow_probability/python/distributions/logitnormal_test.py 127 tensorflow_probability/python/distributions/loglogistic_test.py 177 tensorflow_probability/python/distributions/lognormal_test.py 83 tensorflow_probability/python/distributions/markov_chain_test.py 319 tensorflow_probability/python/distributions/masked_test.py 110 tensorflow_probability/python/distributions/matrix_normal_linear_operator_test.py 164 tensorflow_probability/python/distributions/matrix_t_linear_operator_test.py 159 tensorflow_probability/python/distributions/mixture_same_family_test.py 462 tensorflow_probability/python/distributions/mixture_test.py 780 tensorflow_probability/python/distributions/moyal_test.py 223 tensorflow_probability/python/distributions/multinomial_test.py 389 tensorflow_probability/python/distributions/multivariate_student_t_test.py 389 tensorflow_probability/python/distributions/mvn_diag_plus_low_rank_covariance_test.py 44 tensorflow_probability/python/distributions/mvn_diag_plus_low_rank_test.py 298 tensorflow_probability/python/distributions/mvn_diag_test.py 315 tensorflow_probability/python/distributions/mvn_full_covariance_test.py 123 tensorflow_probability/python/distributions/mvn_linear_operator_test.py 178 tensorflow_probability/python/distributions/mvn_low_rank_update_linear_operator_covariance_test.py 314 tensorflow_probability/python/distributions/mvn_tril_test.py 466 tensorflow_probability/python/distributions/negative_binomial_test.py 315 tensorflow_probability/python/distributions/normal_conjugate_posteriors_test.py 181 tensorflow_probability/python/distributions/normal_inverse_gaussian_test.py 192 tensorflow_probability/python/distributions/normal_test.py 396 tensorflow_probability/python/distributions/numerical_properties_test.py 183 tensorflow_probability/python/distributions/onehot_categorical_test.py 290 tensorflow_probability/python/distributions/ordered_logistic_test.py 145 tensorflow_probability/python/distributions/pareto_test.py 309 tensorflow_probability/python/distributions/pert_test.py 181 tensorflow_probability/python/distributions/pixel_cnn_test.py 202 tensorflow_probability/python/distributions/plackett_luce_test.py 126 tensorflow_probability/python/distributions/platform_compatibility_test.py 376 tensorflow_probability/python/distributions/poisson_lognormal_test.py 129 tensorflow_probability/python/distributions/poisson_test.py 464 tensorflow_probability/python/distributions/power_spherical_test.py 500 tensorflow_probability/python/distributions/probit_bernoulli_test.py 285 tensorflow_probability/python/distributions/quantized_distribution_test.py 327 tensorflow_probability/python/distributions/relaxed_bernoulli_test.py 253 tensorflow_probability/python/distributions/relaxed_onehot_categorical_test.py 271 tensorflow_probability/python/distributions/sample_test.py 444 tensorflow_probability/python/distributions/sigmoid_beta_test.py 213 tensorflow_probability/python/distributions/sinh_arcsinh_test.py 167 tensorflow_probability/python/distributions/skellam_test.py 246 tensorflow_probability/python/distributions/spherical_uniform_test.py 181 tensorflow_probability/python/distributions/stochastic_process_properties_test.py 485 tensorflow_probability/python/distributions/stopping_ratio_logistic_test.py 94 tensorflow_probability/python/distributions/student_t_process_regression_model_test.py 253 tensorflow_probability/python/distributions/student_t_process_test.py 191 tensorflow_probability/python/distributions/student_t_test.py 449 tensorflow_probability/python/distributions/transformed_distribution_test.py 992 tensorflow_probability/python/distributions/triangular_test.py 313 tensorflow_probability/python/distributions/truncated_cauchy_test.py 147 tensorflow_probability/python/distributions/truncated_normal_test.py 469 tensorflow_probability/python/distributions/uniform_test.py 273 tensorflow_probability/python/distributions/variational_gaussian_process_test.py 247 tensorflow_probability/python/distributions/von_mises_fisher_test.py 357 tensorflow_probability/python/distributions/von_mises_test.py 354 tensorflow_probability/python/distributions/weibull_test.py 275 tensorflow_probability/python/distributions/wishart_test.py 421 tensorflow_probability/python/distributions/zipf_test.py 331 tensorflow_probability/python/experimental/auto_batching/allocation_strategy_test.py 62 tensorflow_probability/python/experimental/auto_batching/dsl_test.py 134 tensorflow_probability/python/experimental/auto_batching/frontend_test.py 507 tensorflow_probability/python/experimental/auto_batching/instructions_test.py 58 tensorflow_probability/python/experimental/auto_batching/lowering_test.py 35 tensorflow_probability/python/experimental/auto_batching/numpy_backend_test.py 47 tensorflow_probability/python/experimental/auto_batching/stack_optimization_test.py 18 tensorflow_probability/python/experimental/auto_batching/stackless_test.py 49 tensorflow_probability/python/experimental/auto_batching/test_programs.py 417 tensorflow_probability/python/experimental/auto_batching/tf_backend_test.py 62 tensorflow_probability/python/experimental/auto_batching/type_inference_test.py 152 tensorflow_probability/python/experimental/auto_batching/virtual_machine_test.py 140 tensorflow_probability/python/experimental/bijectors/distribution_bijectors_test.py 160 tensorflow_probability/python/experimental/bijectors/highway_flow_test.py 141 tensorflow_probability/python/experimental/bijectors/scalar_function_with_inferred_inverse_test.py 201 tensorflow_probability/python/experimental/bijectors/sharded_test.py 70 tensorflow_probability/python/experimental/distribute/diagonal_mass_matrix_adaptation_test.py 115 tensorflow_probability/python/experimental/distribute/joint_distribution_test.py 317 tensorflow_probability/python/experimental/distribute/sharded_test.py 355 tensorflow_probability/python/experimental/distributions/importance_resample_test.py 199 tensorflow_probability/python/experimental/distributions/increment_log_prob_test.py 88 tensorflow_probability/python/experimental/distributions/joint_distribution_pinned_test.py 397 tensorflow_probability/python/experimental/distributions/marginal_fns_test.py 123 tensorflow_probability/python/experimental/distributions/mvn_precision_factor_linop_test.py 166 tensorflow_probability/python/experimental/joint_distribution_layers/layers_test.py 193 tensorflow_probability/python/experimental/linalg/linear_operator_psd_kernel_test.py 248 tensorflow_probability/python/experimental/linalg/no_pivot_ldl_test.py 73 tensorflow_probability/python/experimental/marginalize/logeinsumexp_test.py 142 tensorflow_probability/python/experimental/marginalize/marginalizable_test.py 229 tensorflow_probability/python/experimental/math/manual_special_functions_test.py 354 tensorflow_probability/python/experimental/mcmc/covariance_reducer_test.py 100 tensorflow_probability/python/experimental/mcmc/diagonal_mass_matrix_adaptation_test.py 319 tensorflow_probability/python/experimental/mcmc/elliptical_slice_sampler_test.py 95 tensorflow_probability/python/experimental/mcmc/expectations_reducer_test.py 60 tensorflow_probability/python/experimental/mcmc/gradient_based_trajectory_length_adaptation_test.py 590 tensorflow_probability/python/experimental/mcmc/initialization_test.py 140 tensorflow_probability/python/experimental/mcmc/internal/test_fixtures.py 70 tensorflow_probability/python/experimental/mcmc/kernel_builder_test.py 97 tensorflow_probability/python/experimental/mcmc/kernel_outputs_test.py 36 tensorflow_probability/python/experimental/mcmc/nuts_autobatching_test.py 309 tensorflow_probability/python/experimental/mcmc/nuts_autobatching_xla_test.py 92 tensorflow_probability/python/experimental/mcmc/particle_filter_augmentation_test.py 103 tensorflow_probability/python/experimental/mcmc/particle_filter_test.py 439 tensorflow_probability/python/experimental/mcmc/pnuts_test.py 900 tensorflow_probability/python/experimental/mcmc/potential_scale_reduction_reducer_test.py 79 tensorflow_probability/python/experimental/mcmc/preconditioned_hmc_test.py 633 tensorflow_probability/python/experimental/mcmc/progress_bar_reducer_test.py 45 tensorflow_probability/python/experimental/mcmc/sample_discarding_kernel_test.py 249 tensorflow_probability/python/experimental/mcmc/sample_fold_test.py 424 tensorflow_probability/python/experimental/mcmc/sample_sequential_monte_carlo_test.py 205 tensorflow_probability/python/experimental/mcmc/sample_test.py 97 tensorflow_probability/python/experimental/mcmc/sequential_monte_carlo_kernel_test.py 99 tensorflow_probability/python/experimental/mcmc/sharded_test.py 35 tensorflow_probability/python/experimental/mcmc/snaper_hmc_test.py 622 tensorflow_probability/python/experimental/mcmc/step_test.py 122 tensorflow_probability/python/experimental/mcmc/thermodynamic_integrals_test.py 150 tensorflow_probability/python/experimental/mcmc/thinning_kernel_test.py 158 tensorflow_probability/python/experimental/mcmc/tracing_reducer_test.py 59 tensorflow_probability/python/experimental/mcmc/weighted_resampling_test.py 233 tensorflow_probability/python/experimental/mcmc/windowed_sampling_test.py 654 tensorflow_probability/python/experimental/mcmc/with_reductions_test.py 384 tensorflow_probability/python/experimental/nn/affine_layers_test.py 149 tensorflow_probability/python/experimental/nn/convolutional_layers_test.py 113 tensorflow_probability/python/experimental/nn/convolutional_layers_v2_test.py 115 tensorflow_probability/python/experimental/nn/convolutional_transpose_layers_test.py 97 tensorflow_probability/python/experimental/nn/layers_test.py 74 tensorflow_probability/python/experimental/nn/util/convolution_util_test.py 435 tensorflow_probability/python/experimental/nn/util/kernel_bias_test.py 47 tensorflow_probability/python/experimental/nn/util/random_variable_test.py 69 tensorflow_probability/python/experimental/parallel_filter/parallel_kalman_filter_test.py 335 tensorflow_probability/python/experimental/sequential/ensemble_adjustment_kalman_filter_test.py 79 tensorflow_probability/python/experimental/sequential/ensemble_kalman_filter_test.py 180 tensorflow_probability/python/experimental/sequential/extended_kalman_filter_test.py 146 tensorflow_probability/python/experimental/sequential/iterated_filter_test.py 122 tensorflow_probability/python/experimental/stats/sample_stats_test.py 307 tensorflow_probability/python/experimental/sts_gibbs/benchmarks_test.py 68 tensorflow_probability/python/experimental/sts_gibbs/gibbs_sampler_test.py 407 tensorflow_probability/python/experimental/sts_gibbs/spike_and_slab_test.py 235 tensorflow_probability/python/experimental/util/composite_tensor_test.py 378 tensorflow_probability/python/experimental/util/deferred_module_test.py 114 tensorflow_probability/python/experimental/util/jit_public_methods_test.py 93 tensorflow_probability/python/experimental/util/trainable_test.py 284 tensorflow_probability/python/experimental/vi/automatic_structured_vi_test.py 352 tensorflow_probability/python/experimental/vi/surrogate_posteriors_test.py 549 tensorflow_probability/python/experimental/vi/util/trainable_linear_operators_test.py 176 tensorflow_probability/python/glm/family_test.py 222 tensorflow_probability/python/glm/fisher_scoring_test.py 270 tensorflow_probability/python/glm/proximal_hessian_test.py 312 tensorflow_probability/python/internal/auto_composite_tensor_test.py 564 tensorflow_probability/python/internal/backend/numpy/numpy_test.py 1609 tensorflow_probability/python/internal/backend/numpy/rewrite_equivalence_test.py 52 tensorflow_probability/python/internal/backend/numpy/tensor_array_ops_test.py 64 tensorflow_probability/python/internal/backend/numpy/test_lib.py 55 tensorflow_probability/python/internal/batch_shape_lib_test.py 124 tensorflow_probability/python/internal/batched_rejection_sampler_test.py 82 tensorflow_probability/python/internal/broadcast_util_test.py 37 tensorflow_probability/python/internal/cache_util_test.py 252 tensorflow_probability/python/internal/callable_util_test.py 57 tensorflow_probability/python/internal/custom_gradient_test.py 96 tensorflow_probability/python/internal/distribute_lib_test.py 631 tensorflow_probability/python/internal/distribution_util_test.py 840 tensorflow_probability/python/internal/docstring_util_test.py 20 tensorflow_probability/python/internal/dtype_util_test.py 161 tensorflow_probability/python/internal/hypothesis_testlib_test.py 34 tensorflow_probability/python/internal/implementation_selection_test.py 53 tensorflow_probability/python/internal/loop_util_test.py 144 tensorflow_probability/python/internal/nest_util_test.py 435 tensorflow_probability/python/internal/numerics_testing_test.py 27 tensorflow_probability/python/internal/prefer_static_test.py 429 tensorflow_probability/python/internal/samplers_test.py 94 tensorflow_probability/python/internal/slicing_test.py 219 tensorflow_probability/python/internal/special_math_test.py 200 tensorflow_probability/python/internal/structural_tuple_test.py 133 tensorflow_probability/python/internal/tensor_util_test.py 133 tensorflow_probability/python/internal/tensorshape_util_test.py 46 tensorflow_probability/python/internal/test_combinations.py 151 tensorflow_probability/python/internal/test_combinations_test.py 64 tensorflow_probability/python/internal/test_util.py 582 tensorflow_probability/python/internal/test_util_scipy.py 24 tensorflow_probability/python/internal/test_util_test.py 231 tensorflow_probability/python/internal/trainable_state_util_test.py 254 tensorflow_probability/python/internal/unnest_test.py 275 tensorflow_probability/python/internal/vectorization_util_test.py 158 tensorflow_probability/python/layers/conv_variational_test.py 529 tensorflow_probability/python/layers/dense_variational_test.py 397 tensorflow_probability/python/layers/dense_variational_v2_test.py 55 tensorflow_probability/python/layers/distribution_layer_test.py 1194 tensorflow_probability/python/layers/initializers_test.py 22 tensorflow_probability/python/layers/internal/distribution_tensor_coercible_test.py 204 tensorflow_probability/python/layers/internal/tensor_tuple_test.py 71 tensorflow_probability/python/layers/masked_autoregressive_test.py 38 tensorflow_probability/python/layers/variable_input_test.py 61 tensorflow_probability/python/layers/weight_norm_test.py 239 tensorflow_probability/python/math/bessel_test.py 474 tensorflow_probability/python/math/custom_gradient_test.py 79 tensorflow_probability/python/math/diag_jacobian_test.py 73 tensorflow_probability/python/math/generic_test.py 572 tensorflow_probability/python/math/gradient_test.py 234 tensorflow_probability/python/math/gram_schmidt_test.py 80 tensorflow_probability/python/math/hypergeometric_test.py 170 tensorflow_probability/python/math/integration_test.py 150 tensorflow_probability/python/math/interpolation_test.py 653 tensorflow_probability/python/math/linalg_test.py 624 tensorflow_probability/python/math/minimize_test.py 419 tensorflow_probability/python/math/numeric_test.py 56 tensorflow_probability/python/math/ode/bdf_util_test.py 91 tensorflow_probability/python/math/ode/ode_test.py 601 tensorflow_probability/python/math/ode/runge_kutta_util_test.py 114 tensorflow_probability/python/math/ode/util_test.py 63 tensorflow_probability/python/math/ode/xla_test.py 37 tensorflow_probability/python/math/psd_kernels/changepoint_test.py 94 tensorflow_probability/python/math/psd_kernels/exp_sin_squared_test.py 94 tensorflow_probability/python/math/psd_kernels/exponential_curve_test.py 72 tensorflow_probability/python/math/psd_kernels/exponentiated_quadratic_test.py 119 tensorflow_probability/python/math/psd_kernels/feature_scaled_test.py 108 tensorflow_probability/python/math/psd_kernels/feature_transformed_test.py 150 tensorflow_probability/python/math/psd_kernels/internal/util_test.py 145 tensorflow_probability/python/math/psd_kernels/kumaraswamy_transformed_test.py 99 tensorflow_probability/python/math/psd_kernels/matern_test.py 351 tensorflow_probability/python/math/psd_kernels/parabolic_test.py 85 tensorflow_probability/python/math/psd_kernels/pointwise_exponential_test.py 31 tensorflow_probability/python/math/psd_kernels/polynomial_test.py 254 tensorflow_probability/python/math/psd_kernels/positive_semidefinite_kernel_test.py 368 tensorflow_probability/python/math/psd_kernels/psd_kernel_properties_test.py 176 tensorflow_probability/python/math/psd_kernels/rational_quadratic_test.py 137 tensorflow_probability/python/math/psd_kernels/schur_complement_test.py 261 tensorflow_probability/python/math/random_ops_test.py 71 tensorflow_probability/python/math/root_search_test.py 252 tensorflow_probability/python/math/scan_associative_test.py 174 tensorflow_probability/python/math/sparse_test.py 129 tensorflow_probability/python/math/special_test.py 680 tensorflow_probability/python/mcmc/diagnostic_test.py 514 tensorflow_probability/python/mcmc/dual_averaging_step_size_adaptation_test.py 586 tensorflow_probability/python/mcmc/eight_schools_hmc_eager_test.py 10 tensorflow_probability/python/mcmc/eight_schools_hmc_graph_test.py 8 tensorflow_probability/python/mcmc/hmc_test.py 917 tensorflow_probability/python/mcmc/internal/leapfrog_integrator_test.py 53 tensorflow_probability/python/mcmc/internal/util_test.py 341 tensorflow_probability/python/mcmc/langevin_test.py 207 tensorflow_probability/python/mcmc/metropolis_hastings_test.py 297 tensorflow_probability/python/mcmc/nuts_test.py 519 tensorflow_probability/python/mcmc/random_walk_metropolis_test.py 199 tensorflow_probability/python/mcmc/replica_exchange_mc_test.py 1166 tensorflow_probability/python/mcmc/sample_annealed_importance_test.py 148 tensorflow_probability/python/mcmc/sample_halton_sequence_test.py 135 tensorflow_probability/python/mcmc/sample_test.py 366 tensorflow_probability/python/mcmc/simple_step_size_adaptation_test.py 454 tensorflow_probability/python/mcmc/slice_sampler_test.py 207 tensorflow_probability/python/mcmc/transformed_kernel_test.py 403 tensorflow_probability/python/monte_carlo/expectation_test.py 141 tensorflow_probability/python/optimizer/bfgs_test.py 401 tensorflow_probability/python/optimizer/convergence_criteria/loss_not_decreasing_test.py 110 tensorflow_probability/python/optimizer/convergence_criteria/successive_gradients_are_uncorrelated_test.py 53 tensorflow_probability/python/optimizer/differential_evolution_test.py 150 tensorflow_probability/python/optimizer/lbfgs_test.py 366 tensorflow_probability/python/optimizer/linesearch/hager_zhang_test.py 248 tensorflow_probability/python/optimizer/linesearch/internal/hager_zhang_lib_test.py 240 tensorflow_probability/python/optimizer/nelder_mead_test.py 240 tensorflow_probability/python/optimizer/proximal_hessian_sparse_test.py 141 tensorflow_probability/python/optimizer/sgld_test.py 295 tensorflow_probability/python/optimizer/variational_sgd_test.py 278 tensorflow_probability/python/random/random_ops_test.py 129 tensorflow_probability/python/stats/calibration_test.py 253 tensorflow_probability/python/stats/kendalls_tau_test.py 63 tensorflow_probability/python/stats/leave_one_out_test.py 187 tensorflow_probability/python/stats/moving_stats_test.py 105 tensorflow_probability/python/stats/quantiles_test.py 589 tensorflow_probability/python/stats/ranking_test.py 61 tensorflow_probability/python/stats/sample_stats_test.py 509 tensorflow_probability/python/sts/anomaly_detection/anomaly_detection_test.py 136 tensorflow_probability/python/sts/components/autoregressive_integrated_moving_average_test.py 61 tensorflow_probability/python/sts/components/autoregressive_moving_average_test.py 157 tensorflow_probability/python/sts/components/autoregressive_test.py 113 tensorflow_probability/python/sts/components/dynamic_regression_test.py 106 tensorflow_probability/python/sts/components/local_level_test.py 103 tensorflow_probability/python/sts/components/local_linear_trend_test.py 116 tensorflow_probability/python/sts/components/regression_test.py 157 tensorflow_probability/python/sts/components/seasonal_test.py 270 tensorflow_probability/python/sts/components/semilocal_linear_trend_test.py 135 tensorflow_probability/python/sts/components/smooth_seasonal_test.py 108 tensorflow_probability/python/sts/components/sum_test.py 320 tensorflow_probability/python/sts/decomposition_test.py 106 tensorflow_probability/python/sts/default_model_test.py 93 tensorflow_probability/python/sts/fitting_test.py 256 tensorflow_probability/python/sts/forecast_test.py 264 tensorflow_probability/python/sts/holiday_effects_test.py 91 tensorflow_probability/python/sts/internal/missing_values_util_test.py 56 tensorflow_probability/python/sts/internal/seasonality_util_test.py 121 tensorflow_probability/python/sts/internal/util_test.py 278 tensorflow_probability/python/sts/regularization_test.py 173 tensorflow_probability/python/sts/structural_time_series_test.py 420 tensorflow_probability/python/util/deferred_tensor_test.py 738 tensorflow_probability/python/util/seed_stream_test.py 51 tensorflow_probability/python/vi/csiszar_divergence_test.py 836 tensorflow_probability/python/vi/mutual_information_test.py 150 tensorflow_probability/python/vi/optimization_test.py 258 tensorflow_probability/substrates/meta/jax_integration_test.py 37 tensorflow_probability/substrates/meta/numpy_integration_test.py 20 testing/run_github_tests.sh 43 testing/run_tfp_test.sh 18