Searched refs:target_dist (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | rejection_resample_test.py | 43 target_dist = [0.9, 0.05, 0.05, 0.0, 0.0]
52 target_dist=target_dist,
70 self.assertAllClose(target_dist, returned_dist, atol=1e-2)
77 target_dist = [0.5, 0.5] if only_initial_dist else [0.0, 1.0]
89 target_dist=target_dist,
102 target_dist = [0.0, 0.0, 0.0, 1.0]
120 target_dist=target_dist,
135 self.assertAllClose(target_dist, bincount, atol=1e-2)
|
| /external/tensorflow/tensorflow/python/data/experimental/benchmarks/ |
| D | rejection_resample_benchmark.py | 32 target_dist = [0.0, 0.0, 0.0, 1.0]
43 target_dist=target_dist,
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | resampling.py | 37 def rejection_resample(class_func, target_dist, initial_dist=None, seed=None): argument
58 target_dist_t = ops.convert_to_tensor(target_dist, name="target_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…
|