experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update ops-related pbtxt files.

Browse Source 
PiperOrigin-RevId: 236428215
This commit is contained in:
A. Unique TensorFlower 2019-03-01 22:19:25 -08:00 committed by TensorFlower Gardener
parent 876604a93b
commit 0e18a225eb
2 changed files with 254 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

170
tensorflow/core/ops/compat/ops_history.v1.pbtxt
View File

@ -22251,6 +22251,89 @@ op {
} }
is_stateful: true is_stateful: true
} }
op {
name: "EnqueueTPUEmbeddingSparseBatch"
input_arg {
name: "sample_indices"
type_attr: "T1"
number_attr: "N"
}
input_arg {
name: "embedding_indices"
type_attr: "T2"
number_attr: "N"
}
input_arg {
name: "aggregation_weights"
type_attr: "T3"
number_attr: "N"
}
input_arg {
name: "mode_override"
type: DT_STRING
}
attr {
name: "T1"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "T2"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "T3"
type: "type"
default_value {
type: DT_FLOAT
}
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
attr {
name: "N"
type: "int"
has_minimum: true
minimum: 1
}
attr {
name: "device_ordinal"
type: "int"
default_value {
i: -1
}
}
attr {
name: "combiners"
type: "list(string)"
default_value {
list {
}
}
}
is_stateful: true
}
op { op {
name: "EnqueueTPUEmbeddingSparseTensorBatch" name: "EnqueueTPUEmbeddingSparseTensorBatch"
input_arg { input_arg {
@ -22299,6 +22382,93 @@ op {
} }
is_stateful: true is_stateful: true
} }
op {
name: "EnqueueTPUEmbeddingSparseTensorBatch"
input_arg {
name: "sample_indices"
type_attr: "T1"
number_attr: "N"
}
input_arg {
name: "embedding_indices"
type_attr: "T2"
number_attr: "N"
}
input_arg {
name: "aggregation_weights"
type_attr: "T3"
number_attr: "N"
}
input_arg {
name: "mode_override"
type: DT_STRING
}
attr {
name: "T1"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "T2"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "T3"
type: "type"
default_value {
type: DT_FLOAT
}
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
attr {
name: "N"
type: "int"
has_minimum: true
minimum: 1
}
attr {
name: "device_ordinal"
type: "int"
default_value {
i: -1
}
}
attr {
name: "combiners"
type: "list(string)"
default_value {
list {
}
}
}
attr {
name: "table_ids"
type: "list(int)"
}
is_stateful: true
}
op { op {
name: "EnsureShape" name: "EnsureShape"
input_arg { input_arg {

90
tensorflow/core/ops/ops.pbtxt
View File

@ -10421,23 +10421,62 @@ op {
name: "EnqueueTPUEmbeddingSparseBatch" name: "EnqueueTPUEmbeddingSparseBatch"
input_arg { input_arg {
name: "sample_indices" name: "sample_indices"
type: DT_INT32 type_attr: "T1"
number_attr: "N" number_attr: "N"
} }
input_arg { input_arg {
name: "embedding_indices" name: "embedding_indices"
type: DT_INT32 type_attr: "T2"
number_attr: "N" number_attr: "N"
} }
input_arg { input_arg {
name: "aggregation_weights" name: "aggregation_weights"
type: DT_FLOAT type_attr: "T3"
number_attr: "N" number_attr: "N"
} }
input_arg { input_arg {
name: "mode_override" name: "mode_override"
type: DT_STRING type: DT_STRING
} }
attr {
name: "T1"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "T2"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "T3"
type: "type"
default_value {
type: DT_FLOAT
}
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
attr { attr {
name: "N" name: "N"
type: "int" type: "int"
@ -10465,23 +10504,62 @@ op {
name: "EnqueueTPUEmbeddingSparseTensorBatch" name: "EnqueueTPUEmbeddingSparseTensorBatch"
input_arg { input_arg {
name: "sample_indices" name: "sample_indices"
type: DT_INT32 type_attr: "T1"
number_attr: "N" number_attr: "N"
} }
input_arg { input_arg {
name: "embedding_indices" name: "embedding_indices"
type: DT_INT32 type_attr: "T2"
number_attr: "N" number_attr: "N"
} }
input_arg { input_arg {
name: "aggregation_weights" name: "aggregation_weights"
type: DT_FLOAT type_attr: "T3"
number_attr: "N" number_attr: "N"
} }
input_arg { input_arg {
name: "mode_override" name: "mode_override"
type: DT_STRING type: DT_STRING
} }
attr {
name: "T1"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "T2"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "T3"
type: "type"
default_value {
type: DT_FLOAT
}
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
attr { attr {
name: "N" name: "N"
type: "int" type: "int"