Searched refs:initial_dist_ds (Results 1 – 1 of 1) sorted by relevance
66 initial_dist_ds = dataset_ops.Dataset.from_tensors(73 initial_dist_ds = _estimate_initial_dist_ds(target_dist_t,75 acceptance_and_original_prob_ds = initial_dist_ds.map(82 filtered_ds = _filter_ds(dataset, acceptance_dist_ds, initial_dist_ds,132 def _filter_ds(dataset, acceptance_dist_ds, initial_dist_ds, class_func, seed): argument159 initial_dist_ds))191 initial_dist_ds = (class_values_ds.batch(dist_estimation_batch_size)196 return initial_dist_ds