Adding prior distribution computation for HMM

PiperOrigin-RevId: 340521856
bacoco · Nov 3, 2020 · a9b4381 · a9b4381
1 parent 1a66cc4
commit a9b4381
Show file tree

Hide file tree

Showing 3 changed files with 376 additions and 143 deletions.
diff --git a/tensorflow_probability/python/distributions/BUILD b/tensorflow_probability/python/distributions/BUILD
@@ -2558,7 +2558,7 @@ multi_substrate_py_test(
     size = "medium",
     srcs = ["hidden_markov_model_test.py"],
     jax_tags = ["flaky"],
-    shard_count = 4,
+    shard_count = 8,
     deps = [
         # absl/testing:parameterized dep,
         # numpy dep,