xref: /aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/api_def_SamplingDataset.pbtxt (revision b6fb3261f9314811a0f4371741dbb8839866f948) 
1op {
2  graph_op_name: "SamplingDataset"
3  visibility: HIDDEN
4  in_arg {
5    name: "rate"
6    description: <<END
7A scalar representing the sample rate. Each element of `input_dataset` is
8retained with this probability, independent of all other elements.
9END
10  }
11  in_arg {
12    name: "seed"
13    description: <<END
14A scalar representing seed of random number generator.
15END
16  }
17  in_arg {
18    name: "seed2"
19    description: <<END
20A scalar representing seed2 of random number generator.
21END
22  }
23  summary: "Creates a dataset that takes a Bernoulli sample of the contents of another dataset."
24  description: <<END
25There is no transformation in the `tf.data` Python API for creating this dataset.
26Instead, it is created as a result of the `filter_with_random_uniform_fusion`
27static optimization. Whether this optimization is performed is determined by the
28`experimental_optimization.filter_with_random_uniform_fusion` option of
29`tf.data.Options`.
30END
31}
32