Searched refs:initial_dist_t (Results 1 – 1 of 1) sorted by relevance
62 initial_dist_t = ops.convert_to_tensor(initial_dist, name="initial_dist")64 _calculate_acceptance_probs_with_mixing(initial_dist_t,67 initial_dist_t).repeat()88 initial_dist_t, target_dist_t) if initial_dist is not None else None109 def _get_prob_original_static(initial_dist_t, target_dist_t): argument123 init_static = tensor_util.constant_value(initial_dist_t)