initial tests, updated init and build

Author: gisilvs

tensorflow_probability/python/experimental/vi/BUILD

@@ -31,6 +31,7 @@ py_library(
     srcs_version = "PY3",
     deps = [
         ":automatic_structured_vi",
+        ":cascading_flows",
         ":surrogate_posteriors",
         "//tensorflow_probability/python/experimental/vi/util",
         "//tensorflow_probability/python/internal:all_util",
@@ -67,6 +68,37 @@ py_library(
     ],
 )
 
+py_library(
+    name = "cascading_flows",
+    srcs = ["cascading_flows.py.py"],
+    srcs_version = "PY3",
+    deps = [
+        # tensorflow dep,
+
+        "//tensorflow_probability/python/bijectors:build_highway_flow_layer",
+        "//tensorflow_probability/python/bijectors:chain",
+        "//tensorflow_probability/python/bijectors:reshape",
+        "//tensorflow_probability/python/bijectors:scale",
+        "//tensorflow_probability/python/bijectors:shift",
+        "//tensorflow_probability/python/bijectors:split",
+        "//tensorflow_probability/python/distributions:batch_broadcast",
+        "//tensorflow_probability/python/distributions:beta",
+        "//tensorflow_probability/python/distributions:blockwise",
+        "//tensorflow_probability/python/distributions:chi2",
+        "//tensorflow_probability/python/distributions:exponential",
+        "//tensorflow_probability/python/distributions:gamma",
+        "//tensorflow_probability/python/distributions:half_normal",
+        "//tensorflow_probability/python/distributions:joint_distribution_auto_batched",
+        "//tensorflow_probability/python/distributions:joint_distribution_coroutine",
+        "//tensorflow_probability/python/distributions:normal",
+        "//tensorflow_probability/python/distributions:sample",
+        "//tensorflow_probability/python/distributions:transformed_distribution",
+        "//tensorflow_probability/python/distributions:truncated_normal",
+        "//tensorflow_probability/python/distributions:uniform",
+        "//tensorflow_probability/python/internal:samplers",
+    ],
+)
+
 py_library(
     name = "surrogate_posteriors",
     srcs = ["surrogate_posteriors.py"],

tensorflow_probability/python/experimental/vi/__init__.py

@@ -29,6 +29,7 @@
     'build_affine_surrogate_posterior',
     'build_affine_surrogate_posterior_from_base_distribution',
     'build_asvi_surrogate_posterior',
+    'builf_cf_surrogate_posterior'
     'build_factored_surrogate_posterior',
     'build_split_flow_surrogate_posterior',
     'build_trainable_location_scale_distribution',