Searched refs:initial_dist (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | resampling.py | 37 def rejection_resample(class_func, target_dist, initial_dist=None, seed=None): argument
61 if initial_dist is not None:
62 initial_dist_t = ops.convert_to_tensor(initial_dist, name="initial_dist")
88 initial_dist_t, target_dist_t) if initial_dist is not None else None
147 def maybe_warn_on_large_rejection(accept_dist, initial_dist): argument
148 proportion_rejected = math_ops.reduce_sum((1 - accept_dist) * initial_dist)
153 accept_dist, [proportion_rejected, initial_dist, accept_dist],
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | rejection_resample_test.py | 44 initial_dist = [0.2] * 5 if initial_known else None
53 initial_dist=initial_dist,
90 initial_dist=init_dist))
121 initial_dist=init_dist))
|
| /external/tensorflow/tensorflow/python/data/experimental/benchmarks/ |
| D | rejection_resample_benchmark.py | 44 initial_dist=init_dist,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.data.experimental.pbtxt | 185 …argspec: "args=[\'class_func\', \'target_dist\', \'initial_dist\', \'seed\'], varargs=None, keywor…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.data.experimental.pbtxt | 213 …argspec: "args=[\'class_func\', \'target_dist\', \'initial_dist\', \'seed\'], varargs=None, keywor…
|