Searched refs:acceptance_dist_ds (Results 1 – 1 of 1) sorted by relevance
68 acceptance_dist_ds = dataset_ops.Dataset.from_tensors(78 acceptance_dist_ds = acceptance_and_original_prob_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): argument158 acceptance_dist_ds = (dataset_ops.Dataset.zip((acceptance_dist_ds,170 (acceptance_dist_ds, dataset)).map(_gather_and_copy)