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
|
||||
}
|
||||
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 {
|
||||
name: "LoadTPUEmbeddingMDLAdagradLightParameters"
|
||||
input_arg {
|
||||
@ -40646,6 +40732,92 @@ op {
|
||||
}
|
||||
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 {
|
||||
name: "RetrieveTPUEmbeddingMDLAdagradLightParameters"
|
||||
output_arg {
|
||||
|
Loading…
Reference in New Issue
Block a user