STT-tensorflow/tensorflow/tools/api/golden/v1/tensorflow.tpu.experimental.embedding.-adam.pbtxt
Bruce Fontaine 81065dbaba Add TPUEmbedding mid level API.
PiperOrigin-RevId: 312299886
Change-Id: If8e24b080c0fc6d841c0c681aae71d7537b704f8
2020-05-19 10:00:34 -07:00

11 lines
773 B
Plaintext

path: "tensorflow.tpu.experimental.embedding.Adam"
tf_class {
is_instance: "<class \'tensorflow.python.tpu.tpu_embedding_v2_utils.Adam\'>"
is_instance: "<class \'tensorflow.python.tpu.tpu_embedding_v2_utils._Optimizer\'>"
is_instance: "<type \'object\'>"
member_method {
name: "__init__"
argspec: "args=[\'self\', \'learning_rate\', \'beta_1\', \'beta_2\', \'epsilon\', \'lazy_adam\', \'sum_inside_sqrt\', \'use_gradient_accumulation\', \'clip_weight_min\', \'clip_weight_max\', \'weight_decay_factor\', \'multiply_weight_decay_factor_by_learning_rate\', \'slot_variable_creation_fn\'], varargs=None, keywords=None, defaults=[\'0.001\', \'0.9\', \'0.999\', \'1e-07\', \'True\', \'True\', \'True\', \'None\', \'None\', \'None\', \'None\', \'None\'], "
}
}