Update ops-related pbtxt files.

PiperOrigin-RevId: 345140752
Change-Id: I052f0fc7ad0cf2f077068af52af414f5da5ca165
This commit is contained in:
A. Unique TensorFlower 2020-12-01 18:17:54 -08:00 committed by TensorFlower Gardener
parent f27608be09
commit 2527840a36
5 changed files with 344 additions and 0 deletions

View File

@ -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
}

View File

@ -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
}

View File

@ -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
}

View File

@ -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
}

View File

@ -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 {