blob: 36280a5e8e746944dce2a0592c08272aed461638 [file] [log] [blame]
op {
name: "LoadTPUEmbeddingCenteredRMSPropParameters"
input_arg {
name: "parameters"
type: DT_FLOAT
}
input_arg {
name: "ms"
type: DT_FLOAT
}
input_arg {
name: "mom"
type: DT_FLOAT
}
input_arg {
name: "mg"
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
}