Path Lines of Code tensorflow_probability/python/__init__.py 79 tensorflow_probability/python/bijectors/__init__.py 162 tensorflow_probability/python/bijectors/absolute_value.py 38 tensorflow_probability/python/bijectors/ascending.py 40 tensorflow_probability/python/bijectors/batch_normalization.py 106 tensorflow_probability/python/bijectors/bijector.py 914 tensorflow_probability/python/bijectors/bijector_test_util.py 175 tensorflow_probability/python/bijectors/blockwise.py 174 tensorflow_probability/python/bijectors/categorical_to_discrete.py 94 tensorflow_probability/python/bijectors/chain.py 65 tensorflow_probability/python/bijectors/cholesky_outer_product.py 83 tensorflow_probability/python/bijectors/cholesky_to_inv_cholesky.py 60 tensorflow_probability/python/bijectors/composition.py 468 tensorflow_probability/python/bijectors/correlation_cholesky.py 77 tensorflow_probability/python/bijectors/cumsum.py 51 tensorflow_probability/python/bijectors/discrete_cosine_transform.py 36 tensorflow_probability/python/bijectors/exp.py 27 tensorflow_probability/python/bijectors/expm1.py 42 tensorflow_probability/python/bijectors/ffjord.py 128 tensorflow_probability/python/bijectors/fill_scale_tril.py 40 tensorflow_probability/python/bijectors/fill_triangular.py 90 tensorflow_probability/python/bijectors/frechet_cdf.py 105 tensorflow_probability/python/bijectors/generalized_pareto.py 86 tensorflow_probability/python/bijectors/gev_cdf.py 134 tensorflow_probability/python/bijectors/glow.py 398 tensorflow_probability/python/bijectors/gompertz_cdf.py 89 tensorflow_probability/python/bijectors/gumbel_cdf.py 76 tensorflow_probability/python/bijectors/householder.py 41 tensorflow_probability/python/bijectors/hypothesis_testlib.py 402 tensorflow_probability/python/bijectors/identity.py 36 tensorflow_probability/python/bijectors/inline.py 65 tensorflow_probability/python/bijectors/invert.py 90 tensorflow_probability/python/bijectors/iterated_sigmoid_centered.py 62 tensorflow_probability/python/bijectors/joint_map.py 58 tensorflow_probability/python/bijectors/kumaraswamy_cdf.py 89 tensorflow_probability/python/bijectors/lambertw_transform.py 93 tensorflow_probability/python/bijectors/ldj_ratio.py 119 tensorflow_probability/python/bijectors/masked_autoregressive.py 496 tensorflow_probability/python/bijectors/matrix_inverse_tril.py 54 tensorflow_probability/python/bijectors/moyal_cdf.py 81 tensorflow_probability/python/bijectors/normal_cdf.py 44 tensorflow_probability/python/bijectors/ordered.py 45 tensorflow_probability/python/bijectors/pad.py 216 tensorflow_probability/python/bijectors/permute.py 91 tensorflow_probability/python/bijectors/power.py 63 tensorflow_probability/python/bijectors/power_transform.py 69 tensorflow_probability/python/bijectors/rational_quadratic_spline.py 247 tensorflow_probability/python/bijectors/rayleigh_cdf.py 80 tensorflow_probability/python/bijectors/real_nvp.py 174 tensorflow_probability/python/bijectors/reciprocal.py 34 tensorflow_probability/python/bijectors/reshape.py 206 tensorflow_probability/python/bijectors/restructure.py 123 tensorflow_probability/python/bijectors/scale.py 82 tensorflow_probability/python/bijectors/scale_matvec_diag.py 44 tensorflow_probability/python/bijectors/scale_matvec_linear_operator.py 121 tensorflow_probability/python/bijectors/scale_matvec_lu.py 105 tensorflow_probability/python/bijectors/scale_matvec_tril.py 40 tensorflow_probability/python/bijectors/shift.py 40 tensorflow_probability/python/bijectors/shifted_gompertz_cdf.py 99 tensorflow_probability/python/bijectors/sigmoid.py 96 tensorflow_probability/python/bijectors/sinh.py 27 tensorflow_probability/python/bijectors/sinh_arcsinh.py 83 tensorflow_probability/python/bijectors/soft_clip.py 125 tensorflow_probability/python/bijectors/softfloor.py 105 tensorflow_probability/python/bijectors/softmax_centered.py 61 tensorflow_probability/python/bijectors/softplus.py 108 tensorflow_probability/python/bijectors/softsign.py 47 tensorflow_probability/python/bijectors/split.py 308 tensorflow_probability/python/bijectors/square.py 36 tensorflow_probability/python/bijectors/tanh.py 27 tensorflow_probability/python/bijectors/transform_diagonal.py 39 tensorflow_probability/python/bijectors/transpose.py 229 tensorflow_probability/python/bijectors/weibull_cdf.py 97 tensorflow_probability/python/build_defs.bzl 363 tensorflow_probability/python/debugging/__init__.py 4 tensorflow_probability/python/debugging/benchmarking/__init__.py 20 tensorflow_probability/python/debugging/benchmarking/benchmark_tf_function.py 92 tensorflow_probability/python/distributions/__init__.py 263 tensorflow_probability/python/distributions/autoregressive.py 150 tensorflow_probability/python/distributions/batch_broadcast.py 233 tensorflow_probability/python/distributions/batch_concat.py 293 tensorflow_probability/python/distributions/batch_reshape.py 414 tensorflow_probability/python/distributions/bates.py 237 tensorflow_probability/python/distributions/bernoulli.py 174 tensorflow_probability/python/distributions/beta.py 224 tensorflow_probability/python/distributions/beta_binomial.py 140 tensorflow_probability/python/distributions/beta_quotient.py 160 tensorflow_probability/python/distributions/binomial.py 318 tensorflow_probability/python/distributions/blockwise.py 248 tensorflow_probability/python/distributions/categorical.py 242 tensorflow_probability/python/distributions/cauchy.py 122 tensorflow_probability/python/distributions/chi.py 81 tensorflow_probability/python/distributions/chi2.py 139 tensorflow_probability/python/distributions/cholesky_lkj.py 133 tensorflow_probability/python/distributions/cholesky_util.py 30 tensorflow_probability/python/distributions/continuous_bernoulli.py 253 tensorflow_probability/python/distributions/deterministic.py 210 tensorflow_probability/python/distributions/dirichlet.py 185 tensorflow_probability/python/distributions/dirichlet_multinomial.py 166 tensorflow_probability/python/distributions/discrete_rejection_sampling.py 64 tensorflow_probability/python/distributions/distribution.py 879 tensorflow_probability/python/distributions/doublesided_maxwell.py 104 tensorflow_probability/python/distributions/dpp.py 204 tensorflow_probability/python/distributions/empirical.py 186 tensorflow_probability/python/distributions/exp_gamma.py 186 tensorflow_probability/python/distributions/exponential.py 70 tensorflow_probability/python/distributions/exponentially_modified_gaussian.py 131 tensorflow_probability/python/distributions/finite_discrete.py 226 tensorflow_probability/python/distributions/gamma.py 479 tensorflow_probability/python/distributions/gamma_gamma.py 165 tensorflow_probability/python/distributions/gaussian_process.py 326 tensorflow_probability/python/distributions/gaussian_process_regression_model.py 267 tensorflow_probability/python/distributions/generalized_normal.py 169 tensorflow_probability/python/distributions/generalized_pareto.py 173 tensorflow_probability/python/distributions/geometric.py 164 tensorflow_probability/python/distributions/gev.py 121 tensorflow_probability/python/distributions/gumbel.py 83 tensorflow_probability/python/distributions/half_cauchy.py 127 tensorflow_probability/python/distributions/half_normal.py 106 tensorflow_probability/python/distributions/half_student_t.py 179 tensorflow_probability/python/distributions/hidden_markov_model.py 717 tensorflow_probability/python/distributions/horseshoe.py 101 tensorflow_probability/python/distributions/hypothesis_testlib.py 910 tensorflow_probability/python/distributions/independent.py 253 tensorflow_probability/python/distributions/internal/__init__.py 1 tensorflow_probability/python/distributions/internal/correlation_matrix_volumes.py 32 tensorflow_probability/python/distributions/internal/correlation_matrix_volumes_lib.py 75 tensorflow_probability/python/distributions/internal/statistical_testing.py 471 tensorflow_probability/python/distributions/inverse_gamma.py 151 tensorflow_probability/python/distributions/inverse_gaussian.py 197 tensorflow_probability/python/distributions/johnson_su.py 111 tensorflow_probability/python/distributions/joint_distribution.py 921 tensorflow_probability/python/distributions/joint_distribution_auto_batched.py 49 tensorflow_probability/python/distributions/joint_distribution_coroutine.py 47 tensorflow_probability/python/distributions/joint_distribution_named.py 119 tensorflow_probability/python/distributions/joint_distribution_sequential.py 235 tensorflow_probability/python/distributions/joint_distribution_util.py 49 tensorflow_probability/python/distributions/kullback_leibler.py 131 tensorflow_probability/python/distributions/kumaraswamy.py 136 tensorflow_probability/python/distributions/lambertw_f.py 163 tensorflow_probability/python/distributions/laplace.py 132 tensorflow_probability/python/distributions/linear_gaussian_ssm.py 1010 tensorflow_probability/python/distributions/lkj.py 297 tensorflow_probability/python/distributions/log_prob_ratio.py 21 tensorflow_probability/python/distributions/logistic.py 106 tensorflow_probability/python/distributions/logitnormal.py 230 tensorflow_probability/python/distributions/loglogistic.py 105 tensorflow_probability/python/distributions/lognormal.py 80 tensorflow_probability/python/distributions/markov_chain.py 368 tensorflow_probability/python/distributions/masked.py 226 tensorflow_probability/python/distributions/matrix_normal_linear_operator.py 124 tensorflow_probability/python/distributions/matrix_t_linear_operator.py 133 tensorflow_probability/python/distributions/mixture.py 242 tensorflow_probability/python/distributions/mixture_same_family.py 333 tensorflow_probability/python/distributions/moyal.py 90 tensorflow_probability/python/distributions/multinomial.py 178 tensorflow_probability/python/distributions/multivariate_student_t.py 197 tensorflow_probability/python/distributions/mvn_diag.py 102 tensorflow_probability/python/distributions/mvn_diag_plus_low_rank.py 89 tensorflow_probability/python/distributions/mvn_diag_plus_low_rank_covariance.py 73 tensorflow_probability/python/distributions/mvn_full_covariance.py 73 tensorflow_probability/python/distributions/mvn_linear_operator.py 160 tensorflow_probability/python/distributions/mvn_low_rank_update_linear_operator_covariance.py 187 tensorflow_probability/python/distributions/mvn_tril.py 78 tensorflow_probability/python/distributions/negative_binomial.py 174 tensorflow_probability/python/distributions/normal.py 125 tensorflow_probability/python/distributions/normal_conjugate_posteriors.py 68 tensorflow_probability/python/distributions/normal_inverse_gaussian.py 161 tensorflow_probability/python/distributions/onehot_categorical.py 207 tensorflow_probability/python/distributions/ordered_logistic.py 144 tensorflow_probability/python/distributions/pareto.py 174 tensorflow_probability/python/distributions/pert.py 149 tensorflow_probability/python/distributions/pixel_cnn.py 425 tensorflow_probability/python/distributions/plackett_luce.py 141 tensorflow_probability/python/distributions/poisson.py 291 tensorflow_probability/python/distributions/poisson_lognormal.py 215 tensorflow_probability/python/distributions/power_spherical.py 258 tensorflow_probability/python/distributions/probit_bernoulli.py 139 tensorflow_probability/python/distributions/quantized_distribution.py 232 tensorflow_probability/python/distributions/relaxed_bernoulli.py 161 tensorflow_probability/python/distributions/relaxed_onehot_categorical.py 262 tensorflow_probability/python/distributions/sample.py 370 tensorflow_probability/python/distributions/sigmoid_beta.py 116 tensorflow_probability/python/distributions/sinh_arcsinh.py 84 tensorflow_probability/python/distributions/skellam.py 187 tensorflow_probability/python/distributions/spherical_uniform.py 127 tensorflow_probability/python/distributions/stopping_ratio_logistic.py 140 tensorflow_probability/python/distributions/student_t.py 178 tensorflow_probability/python/distributions/student_t_process.py 286 tensorflow_probability/python/distributions/student_t_process_regression_model.py 344 tensorflow_probability/python/distributions/transformed_distribution.py 320 tensorflow_probability/python/distributions/triangular.py 143 tensorflow_probability/python/distributions/truncated_cauchy.py 220 tensorflow_probability/python/distributions/truncated_normal.py 290 tensorflow_probability/python/distributions/uniform.py 143 tensorflow_probability/python/distributions/untestable_distributions.py 1 tensorflow_probability/python/distributions/variational_gaussian_process.py 419 tensorflow_probability/python/distributions/vector_exponential_linear_operator.py 141 tensorflow_probability/python/distributions/von_mises.py 301 tensorflow_probability/python/distributions/von_mises_fisher.py 333 tensorflow_probability/python/distributions/weibull.py 131 tensorflow_probability/python/distributions/wishart.py 344 tensorflow_probability/python/distributions/zipf.py 201 tensorflow_probability/python/experimental/__init__.py 47 tensorflow_probability/python/experimental/auto_batching/__init__.py 42 tensorflow_probability/python/experimental/auto_batching/allocation_strategy.py 93 tensorflow_probability/python/experimental/auto_batching/backend_test_lib.py 93 tensorflow_probability/python/experimental/auto_batching/dsl.py 224 tensorflow_probability/python/experimental/auto_batching/frontend.py 373 tensorflow_probability/python/experimental/auto_batching/gast_util.py 33 tensorflow_probability/python/experimental/auto_batching/instructions.py 599 tensorflow_probability/python/experimental/auto_batching/liveness.py 48 tensorflow_probability/python/experimental/auto_batching/lowering.py 178 tensorflow_probability/python/experimental/auto_batching/numpy_backend.py 159 tensorflow_probability/python/experimental/auto_batching/stack_optimization.py 34 tensorflow_probability/python/experimental/auto_batching/stackless.py 144 tensorflow_probability/python/experimental/auto_batching/tf_backend.py 333 tensorflow_probability/python/experimental/auto_batching/type_inference.py 166 tensorflow_probability/python/experimental/auto_batching/virtual_machine.py 155 tensorflow_probability/python/experimental/auto_batching/xla.py 16 tensorflow_probability/python/experimental/bijectors/__init__.py 16 tensorflow_probability/python/experimental/bijectors/distribution_bijectors.py 85 tensorflow_probability/python/experimental/bijectors/highway_flow.py 210 tensorflow_probability/python/experimental/bijectors/scalar_function_with_inferred_inverse.py 204 tensorflow_probability/python/experimental/bijectors/sharded.py 33 tensorflow_probability/python/experimental/distribute/__init__.py 17 tensorflow_probability/python/experimental/distribute/joint_distribution.py 107 tensorflow_probability/python/experimental/distribute/sharded.py 109 tensorflow_probability/python/experimental/distributions/__init__.py 13 tensorflow_probability/python/experimental/distributions/importance_resample.py 252 tensorflow_probability/python/experimental/distributions/increment_log_prob.py 113 tensorflow_probability/python/experimental/distributions/joint_distribution_pinned.py 257 tensorflow_probability/python/experimental/distributions/marginal_fns.py 91 tensorflow_probability/python/experimental/distributions/mvn_precision_factor_linop.py 119 tensorflow_probability/python/experimental/joint_distribution_layers/__init__.py 3 tensorflow_probability/python/experimental/joint_distribution_layers/layers.py 154 tensorflow_probability/python/experimental/linalg/__init__.py 10 tensorflow_probability/python/experimental/linalg/linear_operator_psd_kernel.py 303 tensorflow_probability/python/experimental/linalg/no_pivot_ldl.py 65 tensorflow_probability/python/experimental/marginalize/__init__.py 6 tensorflow_probability/python/experimental/marginalize/logeinsumexp.py 82 tensorflow_probability/python/experimental/marginalize/marginalizable.py 117 tensorflow_probability/python/experimental/math/__init__.py 16 tensorflow_probability/python/experimental/math/manual_special_functions.py 222 tensorflow_probability/python/experimental/mcmc/__init__.py 120 tensorflow_probability/python/experimental/mcmc/covariance_reducer.py 107 tensorflow_probability/python/experimental/mcmc/diagonal_mass_matrix_adaptation.py 177 tensorflow_probability/python/experimental/mcmc/elliptical_slice_sampler.py 212 tensorflow_probability/python/experimental/mcmc/expectations_reducer.py 80 tensorflow_probability/python/experimental/mcmc/gradient_based_trajectory_length_adaptation.py 536 tensorflow_probability/python/experimental/mcmc/initialization.py 116 tensorflow_probability/python/experimental/mcmc/internal/__init__.py 1 tensorflow_probability/python/experimental/mcmc/kernel_builder.py 317 tensorflow_probability/python/experimental/mcmc/kernel_outputs.py 48 tensorflow_probability/python/experimental/mcmc/nuts_autobatching.py 388 tensorflow_probability/python/experimental/mcmc/particle_filter.py 256 tensorflow_probability/python/experimental/mcmc/particle_filter_augmentation.py 74 tensorflow_probability/python/experimental/mcmc/potential_scale_reduction_reducer.py 69 tensorflow_probability/python/experimental/mcmc/preconditioned_hmc.py 199 tensorflow_probability/python/experimental/mcmc/preconditioned_nuts.py 826 tensorflow_probability/python/experimental/mcmc/preconditioning_utils.py 136 tensorflow_probability/python/experimental/mcmc/progress_bar_reducer.py 47 tensorflow_probability/python/experimental/mcmc/reducer.py 14 tensorflow_probability/python/experimental/mcmc/run.py 94 tensorflow_probability/python/experimental/mcmc/sample.py 94 tensorflow_probability/python/experimental/mcmc/sample_discarding_kernel.py 81 tensorflow_probability/python/experimental/mcmc/sample_fold.py 124 tensorflow_probability/python/experimental/mcmc/sample_sequential_monte_carlo.py 376 tensorflow_probability/python/experimental/mcmc/sequential_monte_carlo_kernel.py 129 tensorflow_probability/python/experimental/mcmc/sharded.py 41 tensorflow_probability/python/experimental/mcmc/snaper_hmc.py 784 tensorflow_probability/python/experimental/mcmc/step.py 46 tensorflow_probability/python/experimental/mcmc/thermodynamic_integrals.py 41 tensorflow_probability/python/experimental/mcmc/thinning_kernel.py 46 tensorflow_probability/python/experimental/mcmc/tracing_reducer.py 82 tensorflow_probability/python/experimental/mcmc/weighted_resampling.py 195 tensorflow_probability/python/experimental/mcmc/windowed_sampling.py 502 tensorflow_probability/python/experimental/mcmc/with_reductions.py 74 tensorflow_probability/python/experimental/nn/__init__.py 42 tensorflow_probability/python/experimental/nn/affine_layers.py 192 tensorflow_probability/python/experimental/nn/convolutional_layers.py 295 tensorflow_probability/python/experimental/nn/convolutional_layers_v2.py 168 tensorflow_probability/python/experimental/nn/convolutional_transpose_layers.py 184 tensorflow_probability/python/experimental/nn/initializers/__init__.py 12 tensorflow_probability/python/experimental/nn/initializers/initializers.py 77 tensorflow_probability/python/experimental/nn/layers.py 122 tensorflow_probability/python/experimental/nn/losses/__init__.py 12 tensorflow_probability/python/experimental/nn/losses/losses.py 35 tensorflow_probability/python/experimental/nn/util/__init__.py 54 tensorflow_probability/python/experimental/nn/util/convolution_util.py 707 tensorflow_probability/python/experimental/nn/util/kernel_bias.py 107 tensorflow_probability/python/experimental/nn/util/random_variable.py 65 tensorflow_probability/python/experimental/nn/util/utils.py 251 tensorflow_probability/python/experimental/nn/variational_base.py 165 tensorflow_probability/python/experimental/parallel_filter/__init__.py 6 tensorflow_probability/python/experimental/parallel_filter/parallel_kalman_filter_lib.py 411 tensorflow_probability/python/experimental/sequential/__init__.py 22 tensorflow_probability/python/experimental/sequential/ensemble_adjustment_kalman_filter.py 80 tensorflow_probability/python/experimental/sequential/ensemble_kalman_filter.py 118 tensorflow_probability/python/experimental/sequential/extended_kalman_filter.py 148 tensorflow_probability/python/experimental/sequential/iterated_filter.py 393 tensorflow_probability/python/experimental/stats/__init__.py 12 tensorflow_probability/python/experimental/stats/sample_stats.py 343 tensorflow_probability/python/experimental/sts_gibbs/__init__.py 1 tensorflow_probability/python/experimental/sts_gibbs/gibbs_sampler.py 441 tensorflow_probability/python/experimental/sts_gibbs/spike_and_slab.py 276 tensorflow_probability/python/experimental/substrates/__init__.py 13 tensorflow_probability/python/experimental/tangent_spaces/__init__.py 14 tensorflow_probability/python/experimental/tangent_spaces/spaces.py 70 tensorflow_probability/python/experimental/util/__init__.py 18 tensorflow_probability/python/experimental/util/composite_tensor.py 163 tensorflow_probability/python/experimental/util/deferred_module.py 25 tensorflow_probability/python/experimental/util/jit_public_methods.py 77 tensorflow_probability/python/experimental/util/special_methods.py 268 tensorflow_probability/python/experimental/util/trainable.py 73 tensorflow_probability/python/experimental/vi/__init__.py 28 tensorflow_probability/python/experimental/vi/automatic_structured_vi.py 305 tensorflow_probability/python/experimental/vi/surrogate_posteriors.py 273 tensorflow_probability/python/experimental/vi/util/__init__.py 14 tensorflow_probability/python/experimental/vi/util/trainable_linear_operators.py 179 tensorflow_probability/python/glm/__init__.py 46 tensorflow_probability/python/glm/family.py 276 tensorflow_probability/python/glm/fisher_scoring.py 280 tensorflow_probability/python/glm/proximal_hessian.py 117 tensorflow_probability/python/internal/__init__.py 1 tensorflow_probability/python/internal/all_util.py 13 tensorflow_probability/python/internal/assert_util.py 51 tensorflow_probability/python/internal/auto_composite_tensor.py 302 tensorflow_probability/python/internal/backend/__init__.py 1 tensorflow_probability/python/internal/backend/jax/rewrite.py 39 tensorflow_probability/python/internal/backend/meta/gen_linear_operators.py 182 tensorflow_probability/python/internal/backend/meta/gen_tensor_shape.py 44 tensorflow_probability/python/internal/backend/numpy/__init__.py 39 tensorflow_probability/python/internal/backend/numpy/__internal__.py 3 tensorflow_probability/python/internal/backend/numpy/_utils.py 82 tensorflow_probability/python/internal/backend/numpy/bitwise.py 12 tensorflow_probability/python/internal/backend/numpy/compat.py 10 tensorflow_probability/python/internal/backend/numpy/composite_tensor.py 3 tensorflow_probability/python/internal/backend/numpy/config.py 13 tensorflow_probability/python/internal/backend/numpy/control_flow.py 139 tensorflow_probability/python/internal/backend/numpy/data_structures.py 3 tensorflow_probability/python/internal/backend/numpy/debugging.py 227 tensorflow_probability/python/internal/backend/numpy/deprecation.py 23 tensorflow_probability/python/internal/backend/numpy/dtype.py 59 tensorflow_probability/python/internal/backend/numpy/errors.py 8 tensorflow_probability/python/internal/backend/numpy/functional_ops.py 124 tensorflow_probability/python/internal/backend/numpy/gen/__init__.py 1 tensorflow_probability/python/internal/backend/numpy/gen/adjoint_registrations.py 101 tensorflow_probability/python/internal/backend/numpy/gen/cholesky_registrations.py 78 tensorflow_probability/python/internal/backend/numpy/gen/inverse_registrations.py 141 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator.py 599 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_addition.py 172 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_adjoint.py 120 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_algebra.py 139 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_block_diag.py 369 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_block_lower_triangular.py 396 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_circulant.py 363 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_composition.py 136 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_diag.py 134 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_full_matrix.py 81 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_householder.py 137 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_identity.py 372 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_inversion.py 105 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_kronecker.py 301 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_low_rank_update.py 261 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_lower_triangular.py 93 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_toeplitz.py 142 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_util.py 247 tensorflow_probability/python/internal/backend/numpy/gen/linear_operator_zeros.py 260 tensorflow_probability/python/internal/backend/numpy/gen/matmul_registrations.py 183 tensorflow_probability/python/internal/backend/numpy/gen/registrations_util.py 47 tensorflow_probability/python/internal/backend/numpy/gen/solve_registrations.py 163 tensorflow_probability/python/internal/backend/numpy/gen/tensor_shape.py 464 tensorflow_probability/python/internal/backend/numpy/initializers.py 14 tensorflow_probability/python/internal/backend/numpy/keras_layers.py 5 tensorflow_probability/python/internal/backend/numpy/linalg.py 80 tensorflow_probability/python/internal/backend/numpy/linalg_impl.py 349 tensorflow_probability/python/internal/backend/numpy/misc.py 132 tensorflow_probability/python/internal/backend/numpy/nest.py 199 tensorflow_probability/python/internal/backend/numpy/nested_structure_coder.py 3 tensorflow_probability/python/internal/backend/numpy/nn.py 163 tensorflow_probability/python/internal/backend/numpy/numpy_array.py 370 tensorflow_probability/python/internal/backend/numpy/numpy_keras.py 4 tensorflow_probability/python/internal/backend/numpy/numpy_logging.py 103 tensorflow_probability/python/internal/backend/numpy/numpy_math.py 717 tensorflow_probability/python/internal/backend/numpy/numpy_signal.py 73 tensorflow_probability/python/internal/backend/numpy/ops.py 480 tensorflow_probability/python/internal/backend/numpy/private.py 76 tensorflow_probability/python/internal/backend/numpy/random_generators.py 228 tensorflow_probability/python/internal/backend/numpy/raw_ops.py 63 tensorflow_probability/python/internal/backend/numpy/resource_variable_ops.py 7 tensorflow_probability/python/internal/backend/numpy/sets_lib.py 16 tensorflow_probability/python/internal/backend/numpy/sparse_lib.py 15 tensorflow_probability/python/internal/backend/numpy/tensor_array_ops.py 129 tensorflow_probability/python/internal/backend/numpy/tensor_spec.py 9 tensorflow_probability/python/internal/backend/numpy/tf_inspect.py 26 tensorflow_probability/python/internal/backend/numpy/type_spec.py 20 tensorflow_probability/python/internal/backend/numpy/v1.py 119 tensorflow_probability/python/internal/backend/numpy/v2.py 93 tensorflow_probability/python/internal/backend/numpy/variables.py 3 tensorflow_probability/python/internal/batch_shape_lib.py 163 tensorflow_probability/python/internal/batched_rejection_sampler.py 53 tensorflow_probability/python/internal/broadcast_util.py 47 tensorflow_probability/python/internal/cache_util.py 242 tensorflow_probability/python/internal/callable_util.py 22 tensorflow_probability/python/internal/custom_gradient.py 72 tensorflow_probability/python/internal/distribute_lib.py 187 tensorflow_probability/python/internal/distribute_test_lib.py 71 tensorflow_probability/python/internal/distribution_util.py 684 tensorflow_probability/python/internal/docstring_util.py 15 tensorflow_probability/python/internal/dtype_util.py 173 tensorflow_probability/python/internal/hypothesis_testlib.py 538 tensorflow_probability/python/internal/implementation_selection.py 64 tensorflow_probability/python/internal/lazy_loader.py 27 tensorflow_probability/python/internal/loop_util.py 112 tensorflow_probability/python/internal/monte_carlo.py 45 tensorflow_probability/python/internal/name_util.py 65 tensorflow_probability/python/internal/nest_util.py 213 tensorflow_probability/python/internal/numerics_testing.py 109 tensorflow_probability/python/internal/parameter_properties.py 105 tensorflow_probability/python/internal/prefer_static.py 329 tensorflow_probability/python/internal/reparameterization.py 16 tensorflow_probability/python/internal/samplers.py 156 tensorflow_probability/python/internal/slicing.py 104 tensorflow_probability/python/internal/special_math.py 84 tensorflow_probability/python/internal/structural_tuple.py 100 tensorflow_probability/python/internal/tensor_util.py 55 tensorflow_probability/python/internal/tensorshape_util.py 76 tensorflow_probability/python/internal/trainable_state_util.py 138 tensorflow_probability/python/internal/unnest.py 117 tensorflow_probability/python/internal/variadic_reduce.py 120 tensorflow_probability/python/internal/vectorization_util.py 145 tensorflow_probability/python/layers/__init__.py 61 tensorflow_probability/python/layers/conv_variational.py 652 tensorflow_probability/python/layers/dense_variational.py 330 tensorflow_probability/python/layers/dense_variational_v2.py 91 tensorflow_probability/python/layers/distribution_layer.py 806 tensorflow_probability/python/layers/initializers.py 73 tensorflow_probability/python/layers/internal/__init__.py 1 tensorflow_probability/python/layers/internal/distribution_tensor_coercible.py 126 tensorflow_probability/python/layers/internal/tensor_tuple.py 53 tensorflow_probability/python/layers/masked_autoregressive.py 26 tensorflow_probability/python/layers/util.py 96 tensorflow_probability/python/layers/variable_input.py 37 tensorflow_probability/python/layers/weight_norm.py 82 tensorflow_probability/python/math/__init__.py 145 tensorflow_probability/python/math/bessel.py 692 tensorflow_probability/python/math/custom_gradient.py 40 tensorflow_probability/python/math/diag_jacobian.py 97 tensorflow_probability/python/math/generic.py 250 tensorflow_probability/python/math/gradient.py 240 tensorflow_probability/python/math/gram_schmidt.py 23 tensorflow_probability/python/math/hypergeometric.py 281 tensorflow_probability/python/math/integration.py 49 tensorflow_probability/python/math/interpolation.py 392 tensorflow_probability/python/math/linalg.py 468 tensorflow_probability/python/math/minimize.py 214 tensorflow_probability/python/math/numeric.py 23 tensorflow_probability/python/math/ode/__init__.py 14 tensorflow_probability/python/math/ode/base.py 289 tensorflow_probability/python/math/ode/bdf.py 582 tensorflow_probability/python/math/ode/bdf_util.py 188 tensorflow_probability/python/math/ode/dormand_prince.py 433 tensorflow_probability/python/math/ode/runge_kutta_util.py 117 tensorflow_probability/python/math/ode/util.py 110 tensorflow_probability/python/math/psd_kernels/__init__.py 44 tensorflow_probability/python/math/psd_kernels/changepoint.py 156 tensorflow_probability/python/math/psd_kernels/exp_sin_squared.py 86 tensorflow_probability/python/math/psd_kernels/exponential_curve.py 64 tensorflow_probability/python/math/psd_kernels/exponentiated_quadratic.py 106 tensorflow_probability/python/math/psd_kernels/feature_scaled.py 74 tensorflow_probability/python/math/psd_kernels/feature_transformed.py 45 tensorflow_probability/python/math/psd_kernels/hypothesis_testlib.py 561 tensorflow_probability/python/math/psd_kernels/internal/__init__.py 1 tensorflow_probability/python/math/psd_kernels/internal/util.py 99 tensorflow_probability/python/math/psd_kernels/kumaraswamy_transformed.py 85 tensorflow_probability/python/math/psd_kernels/matern.py 276 tensorflow_probability/python/math/psd_kernels/parabolic.py 85 tensorflow_probability/python/math/psd_kernels/pointwise_exponential.py 28 tensorflow_probability/python/math/psd_kernels/polynomial.py 183 tensorflow_probability/python/math/psd_kernels/positive_semidefinite_kernel.py 378 tensorflow_probability/python/math/psd_kernels/rational_quadratic.py 127 tensorflow_probability/python/math/psd_kernels/schur_complement.py 227 tensorflow_probability/python/math/root_search.py 251 tensorflow_probability/python/math/scan_associative.py 150 tensorflow_probability/python/math/sparse.py 19 tensorflow_probability/python/math/special.py 1004 tensorflow_probability/python/mcmc/__init__.py 54 tensorflow_probability/python/mcmc/diagnostic.py 200 tensorflow_probability/python/mcmc/dual_averaging_step_size_adaptation.py 301 tensorflow_probability/python/mcmc/eight_schools_hmc.py 106 tensorflow_probability/python/mcmc/hmc.py 360 tensorflow_probability/python/mcmc/internal/__init__.py 1 tensorflow_probability/python/mcmc/internal/leapfrog_integrator.py 155 tensorflow_probability/python/mcmc/internal/slice_sampler_utils.py 151 tensorflow_probability/python/mcmc/internal/util.py 286 tensorflow_probability/python/mcmc/kernel.py 25 tensorflow_probability/python/mcmc/langevin.py 451 tensorflow_probability/python/mcmc/metropolis_hastings.py 131 tensorflow_probability/python/mcmc/nuts.py 820 tensorflow_probability/python/mcmc/random_walk_metropolis.py 207 tensorflow_probability/python/mcmc/replica_exchange_mc.py 486 tensorflow_probability/python/mcmc/sample.py 101 tensorflow_probability/python/mcmc/sample_annealed_importance.py 112 tensorflow_probability/python/mcmc/sample_halton_sequence.py 105 tensorflow_probability/python/mcmc/simple_step_size_adaptation.py 193 tensorflow_probability/python/mcmc/slice_sampler_kernel.py 259 tensorflow_probability/python/mcmc/transformed_kernel.py 187 tensorflow_probability/python/monte_carlo/__init__.py 6 tensorflow_probability/python/monte_carlo/expectation.py 25 tensorflow_probability/python/optimizer/__init__.py 32 tensorflow_probability/python/optimizer/bfgs.py 161 tensorflow_probability/python/optimizer/bfgs_utils.py 159 tensorflow_probability/python/optimizer/convergence_criteria/__init__.py 10 tensorflow_probability/python/optimizer/convergence_criteria/convergence_criterion.py 34 tensorflow_probability/python/optimizer/convergence_criteria/loss_not_decreasing.py 63 tensorflow_probability/python/optimizer/convergence_criteria/successive_gradients_are_uncorrelated.py 25 tensorflow_probability/python/optimizer/differential_evolution.py 318 tensorflow_probability/python/optimizer/lbfgs.py 143 tensorflow_probability/python/optimizer/linesearch/__init__.py 6 tensorflow_probability/python/optimizer/linesearch/hager_zhang.py 219 tensorflow_probability/python/optimizer/linesearch/internal/__init__.py 1 tensorflow_probability/python/optimizer/linesearch/internal/hager_zhang_lib.py 285 tensorflow_probability/python/optimizer/nelder_mead.py 450 tensorflow_probability/python/optimizer/proximal_hessian_sparse.py 196 tensorflow_probability/python/optimizer/sgld.py 124 tensorflow_probability/python/optimizer/variational_sgd.py 153 tensorflow_probability/python/random/__init__.py 14 tensorflow_probability/python/random/random_ops.py 76 tensorflow_probability/python/stats/__init__.py 64 tensorflow_probability/python/stats/calibration.py 203 tensorflow_probability/python/stats/kendalls_tau.py 137 tensorflow_probability/python/stats/leave_one_out.py 82 tensorflow_probability/python/stats/moving_stats.py 99 tensorflow_probability/python/stats/quantiles.py 359 tensorflow_probability/python/stats/ranking.py 129 tensorflow_probability/python/stats/sample_stats.py 385 tensorflow_probability/python/sts/__init__.py 55 tensorflow_probability/python/sts/anomaly_detection/__init__.py 10 tensorflow_probability/python/sts/anomaly_detection/anomaly_detection_lib.py 214 tensorflow_probability/python/sts/components/__init__.py 50 tensorflow_probability/python/sts/components/autoregressive.py 155 tensorflow_probability/python/sts/components/autoregressive_integrated_moving_average.py 228 tensorflow_probability/python/sts/components/autoregressive_moving_average.py 96 tensorflow_probability/python/sts/components/dynamic_regression.py 122 tensorflow_probability/python/sts/components/local_level.py 140 tensorflow_probability/python/sts/components/local_linear_trend.py 183 tensorflow_probability/python/sts/components/regression.py 183 tensorflow_probability/python/sts/components/seasonal.py 330 tensorflow_probability/python/sts/components/semilocal_linear_trend.py 187 tensorflow_probability/python/sts/components/smooth_seasonal.py 170 tensorflow_probability/python/sts/components/sum.py 212 tensorflow_probability/python/sts/decomposition.py 77 tensorflow_probability/python/sts/default_model.py 40 tensorflow_probability/python/sts/fitting.py 129 tensorflow_probability/python/sts/forecast.py 141 tensorflow_probability/python/sts/holiday_effects.py 61 tensorflow_probability/python/sts/internal/__init__.py 1 tensorflow_probability/python/sts/internal/missing_values_util.py 52 tensorflow_probability/python/sts/internal/seasonality_util.py 123 tensorflow_probability/python/sts/internal/util.py 122 tensorflow_probability/python/sts/regularization.py 120 tensorflow_probability/python/sts/structural_time_series.py 219 tensorflow_probability/python/util/__init__.py 12 tensorflow_probability/python/util/deferred_tensor.py 515 tensorflow_probability/python/util/seed_stream.py 39 tensorflow_probability/python/version.py 11 tensorflow_probability/python/vi/__init__.py 50 tensorflow_probability/python/vi/csiszar_divergence.py 282 tensorflow_probability/python/vi/mutual_information.py 151 tensorflow_probability/python/vi/optimization.py 95