Update ops-related pbtxt files.
PiperOrigin-RevId: 345140752 Change-Id: I052f0fc7ad0cf2f077068af52af414f5da5ca165
This commit is contained in:
parent
f27608be09
commit
2527840a36
@ -0,0 +1,41 @@
|
|||||||
|
op {
|
||||||
|
name: "LoadTPUEmbeddingFrequencyEstimatorParameters"
|
||||||
|
input_arg {
|
||||||
|
name: "parameters"
|
||||||
|
type: DT_FLOAT
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "last_hit_step"
|
||||||
|
type: DT_FLOAT
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "table_id"
|
||||||
|
type: "int"
|
||||||
|
default_value {
|
||||||
|
i: -1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "table_name"
|
||||||
|
type: "string"
|
||||||
|
default_value {
|
||||||
|
s: ""
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "num_shards"
|
||||||
|
type: "int"
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "shard_id"
|
||||||
|
type: "int"
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "config"
|
||||||
|
type: "string"
|
||||||
|
default_value {
|
||||||
|
s: ""
|
||||||
|
}
|
||||||
|
}
|
||||||
|
is_stateful: true
|
||||||
|
}
|
@ -0,0 +1,45 @@
|
|||||||
|
op {
|
||||||
|
name: "LoadTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug"
|
||||||
|
input_arg {
|
||||||
|
name: "parameters"
|
||||||
|
type: DT_FLOAT
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "last_hit_step"
|
||||||
|
type: DT_FLOAT
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "gradient_accumulators"
|
||||||
|
type: DT_FLOAT
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "table_id"
|
||||||
|
type: "int"
|
||||||
|
default_value {
|
||||||
|
i: -1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "table_name"
|
||||||
|
type: "string"
|
||||||
|
default_value {
|
||||||
|
s: ""
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "num_shards"
|
||||||
|
type: "int"
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "shard_id"
|
||||||
|
type: "int"
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "config"
|
||||||
|
type: "string"
|
||||||
|
default_value {
|
||||||
|
s: ""
|
||||||
|
}
|
||||||
|
}
|
||||||
|
is_stateful: true
|
||||||
|
}
|
@ -0,0 +1,41 @@
|
|||||||
|
op {
|
||||||
|
name: "RetrieveTPUEmbeddingFrequencyEstimatorParameters"
|
||||||
|
output_arg {
|
||||||
|
name: "parameters"
|
||||||
|
type: DT_FLOAT
|
||||||
|
}
|
||||||
|
output_arg {
|
||||||
|
name: "last_hit_step"
|
||||||
|
type: DT_FLOAT
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "table_id"
|
||||||
|
type: "int"
|
||||||
|
default_value {
|
||||||
|
i: -1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "table_name"
|
||||||
|
type: "string"
|
||||||
|
default_value {
|
||||||
|
s: ""
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "num_shards"
|
||||||
|
type: "int"
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "shard_id"
|
||||||
|
type: "int"
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "config"
|
||||||
|
type: "string"
|
||||||
|
default_value {
|
||||||
|
s: ""
|
||||||
|
}
|
||||||
|
}
|
||||||
|
is_stateful: true
|
||||||
|
}
|
@ -0,0 +1,45 @@
|
|||||||
|
op {
|
||||||
|
name: "RetrieveTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug"
|
||||||
|
output_arg {
|
||||||
|
name: "parameters"
|
||||||
|
type: DT_FLOAT
|
||||||
|
}
|
||||||
|
output_arg {
|
||||||
|
name: "last_hit_step"
|
||||||
|
type: DT_FLOAT
|
||||||
|
}
|
||||||
|
output_arg {
|
||||||
|
name: "gradient_accumulators"
|
||||||
|
type: DT_FLOAT
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "table_id"
|
||||||
|
type: "int"
|
||||||
|
default_value {
|
||||||
|
i: -1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "table_name"
|
||||||
|
type: "string"
|
||||||
|
default_value {
|
||||||
|
s: ""
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "num_shards"
|
||||||
|
type: "int"
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "shard_id"
|
||||||
|
type: "int"
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "config"
|
||||||
|
type: "string"
|
||||||
|
default_value {
|
||||||
|
s: ""
|
||||||
|
}
|
||||||
|
}
|
||||||
|
is_stateful: true
|
||||||
|
}
|
@ -21595,6 +21595,92 @@ op {
|
|||||||
}
|
}
|
||||||
is_stateful: true
|
is_stateful: true
|
||||||
}
|
}
|
||||||
|
op {
|
||||||
|
name: "LoadTPUEmbeddingFrequencyEstimatorParameters"
|
||||||
|
input_arg {
|
||||||
|
name: "parameters"
|
||||||
|
type: DT_FLOAT
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "last_hit_step"
|
||||||
|
type: DT_FLOAT
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "table_id"
|
||||||
|
type: "int"
|
||||||
|
default_value {
|
||||||
|
i: -1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "table_name"
|
||||||
|
type: "string"
|
||||||
|
default_value {
|
||||||
|
s: ""
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "num_shards"
|
||||||
|
type: "int"
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "shard_id"
|
||||||
|
type: "int"
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "config"
|
||||||
|
type: "string"
|
||||||
|
default_value {
|
||||||
|
s: ""
|
||||||
|
}
|
||||||
|
}
|
||||||
|
is_stateful: true
|
||||||
|
}
|
||||||
|
op {
|
||||||
|
name: "LoadTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug"
|
||||||
|
input_arg {
|
||||||
|
name: "parameters"
|
||||||
|
type: DT_FLOAT
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "last_hit_step"
|
||||||
|
type: DT_FLOAT
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "gradient_accumulators"
|
||||||
|
type: DT_FLOAT
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "table_id"
|
||||||
|
type: "int"
|
||||||
|
default_value {
|
||||||
|
i: -1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "table_name"
|
||||||
|
type: "string"
|
||||||
|
default_value {
|
||||||
|
s: ""
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "num_shards"
|
||||||
|
type: "int"
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "shard_id"
|
||||||
|
type: "int"
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "config"
|
||||||
|
type: "string"
|
||||||
|
default_value {
|
||||||
|
s: ""
|
||||||
|
}
|
||||||
|
}
|
||||||
|
is_stateful: true
|
||||||
|
}
|
||||||
op {
|
op {
|
||||||
name: "LoadTPUEmbeddingMDLAdagradLightParameters"
|
name: "LoadTPUEmbeddingMDLAdagradLightParameters"
|
||||||
input_arg {
|
input_arg {
|
||||||
@ -40646,6 +40732,92 @@ op {
|
|||||||
}
|
}
|
||||||
is_stateful: true
|
is_stateful: true
|
||||||
}
|
}
|
||||||
|
op {
|
||||||
|
name: "RetrieveTPUEmbeddingFrequencyEstimatorParameters"
|
||||||
|
output_arg {
|
||||||
|
name: "parameters"
|
||||||
|
type: DT_FLOAT
|
||||||
|
}
|
||||||
|
output_arg {
|
||||||
|
name: "last_hit_step"
|
||||||
|
type: DT_FLOAT
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "table_id"
|
||||||
|
type: "int"
|
||||||
|
default_value {
|
||||||
|
i: -1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "table_name"
|
||||||
|
type: "string"
|
||||||
|
default_value {
|
||||||
|
s: ""
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "num_shards"
|
||||||
|
type: "int"
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "shard_id"
|
||||||
|
type: "int"
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "config"
|
||||||
|
type: "string"
|
||||||
|
default_value {
|
||||||
|
s: ""
|
||||||
|
}
|
||||||
|
}
|
||||||
|
is_stateful: true
|
||||||
|
}
|
||||||
|
op {
|
||||||
|
name: "RetrieveTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug"
|
||||||
|
output_arg {
|
||||||
|
name: "parameters"
|
||||||
|
type: DT_FLOAT
|
||||||
|
}
|
||||||
|
output_arg {
|
||||||
|
name: "last_hit_step"
|
||||||
|
type: DT_FLOAT
|
||||||
|
}
|
||||||
|
output_arg {
|
||||||
|
name: "gradient_accumulators"
|
||||||
|
type: DT_FLOAT
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "table_id"
|
||||||
|
type: "int"
|
||||||
|
default_value {
|
||||||
|
i: -1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "table_name"
|
||||||
|
type: "string"
|
||||||
|
default_value {
|
||||||
|
s: ""
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "num_shards"
|
||||||
|
type: "int"
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "shard_id"
|
||||||
|
type: "int"
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "config"
|
||||||
|
type: "string"
|
||||||
|
default_value {
|
||||||
|
s: ""
|
||||||
|
}
|
||||||
|
}
|
||||||
|
is_stateful: true
|
||||||
|
}
|
||||||
op {
|
op {
|
||||||
name: "RetrieveTPUEmbeddingMDLAdagradLightParameters"
|
name: "RetrieveTPUEmbeddingMDLAdagradLightParameters"
|
||||||
output_arg {
|
output_arg {
|
||||||
|
Loading…
Reference in New Issue
Block a user