STT-tensorflow/tensorflow/tools/api/golden/v1/tensorflow.random.experimental.pbtxt
Peng Wang aba2ca4603 Adds tf.random.experimental.stateless_split and tf.random.experimental.stateless_fold_in to manage seeds for stateless RNGs.
PiperOrigin-RevId: 302520177
Change-Id: I43c3d2e2aa5dd26a64e978027df0ecb1f6095140
2020-03-23 14:42:43 -07:00

32 lines
863 B
Plaintext

path: "tensorflow.random.experimental"
tf_module {
member {
name: "Algorithm"
mtype: "<class \'enum.EnumMeta\'>"
}
member {
name: "Generator"
mtype: "<type \'type\'>"
}
member_method {
name: "create_rng_state"
argspec: "args=[\'seed\', \'alg\'], varargs=None, keywords=None, defaults=None"
}
member_method {
name: "get_global_generator"
argspec: "args=[], varargs=None, keywords=None, defaults=None"
}
member_method {
name: "set_global_generator"
argspec: "args=[\'generator\'], varargs=None, keywords=None, defaults=None"
}
member_method {
name: "stateless_fold_in"
argspec: "args=[\'seed\', \'data\'], varargs=None, keywords=None, defaults=None"
}
member_method {
name: "stateless_split"
argspec: "args=[\'seed\', \'num\'], varargs=None, keywords=None, defaults=[\'2\'], "
}
}