Commit 047e9bd0 authored by Chris Jewell's avatar Chris Jewell
Browse files

Replace XLA compile

parent 588d479e
......@@ -236,7 +236,7 @@ def trace_results_fn(_, results):
return recurse(f, results)
@tf.function # (autograph=False, experimental_compile=True)
@tf.function(autograph=False, experimental_compile=True)
def sample(n_samples, init_state, sigma_theta, sigma_xi, num_event_updates):
with tf.name_scope("main_mcmc_sample_loop"):
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment