op { | |
name: "LoadTPUEmbeddingAdagradParametersGradAccumDebug" | |
input_arg { | |
name: "parameters" | |
type: DT_FLOAT | |
} | |
input_arg { | |
name: "accumulators" | |
type: DT_FLOAT | |
} | |
input_arg { | |
name: "gradient_accumulators" | |
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 | |
} |