blob: 2867fdaea952e2c5e6cabb969377dc4dbbcf67e8 [file] [log] [blame]
op {
name: "LoadTPUEmbeddingRMSPropParameters"
input_arg {
name: "parameters"
type: DT_FLOAT
}
input_arg {
name: "ms"
type: DT_FLOAT
}
input_arg {
name: "mom"
type: DT_FLOAT
}
attr {
name: "table_id"
type: "int"
default_value {
i: -1
}
has_minimum: true
minimum: -1
}
attr {
name: "table_name"
type: "string"
default_value {
s: ""
}
}
attr {
name: "num_shards"
type: "int"
}
attr {
name: "shard_id"
type: "int"
}
is_stateful: true
}