Lines Matching defs:kernel
19 @@kernel
42 "kernel",
289 return [iter_ + 1] + list(kernel(
522 return [iter_ + 1, ais_weights] + list(kernel(
551 def kernel(target_log_prob_fn,
593 new_x, other_results = hmc.kernel(
694 hmc.kernel(target_log_prob_fn=joint_log_prob,
1048 """Helper to `kernel` which computes the energy change."""
1099 """Helper to `kernel` which expand_dims `is_accepted` to apply tf.where."""
