Update ops-related pbtxt files.

PiperOrigin-RevId: 243117391
This commit is contained in:
A. Unique TensorFlower 2019-04-11 12:18:07 -07:00 committed by TensorFlower Gardener
parent 9a4141c306
commit f8c7522bb4
2 changed files with 167 additions and 0 deletions

View File

@ -13459,6 +13459,63 @@ op {
}
}
}
op {
name: "CombinedNonMaxSuppression"
input_arg {
name: "boxes"
type: DT_FLOAT
}
input_arg {
name: "scores"
type: DT_FLOAT
}
input_arg {
name: "max_output_size_per_class"
type: DT_INT32
}
input_arg {
name: "max_total_size"
type: DT_INT32
}
input_arg {
name: "iou_threshold"
type: DT_FLOAT
}
input_arg {
name: "score_threshold"
type: DT_FLOAT
}
output_arg {
name: "nmsed_boxes"
type: DT_FLOAT
}
output_arg {
name: "nmsed_scores"
type: DT_FLOAT
}
output_arg {
name: "nmsed_classes"
type: DT_FLOAT
}
output_arg {
name: "valid_detections"
type: DT_INT32
}
attr {
name: "pad_per_class"
type: "bool"
default_value {
b: false
}
}
attr {
name: "clip_boxes"
type: "bool"
default_value {
b: true
}
}
}
op {
name: "CompareAndBitpack"
input_arg {
@ -23114,6 +23171,101 @@ op {
}
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)"
}
attr {
name: "max_sequence_lengths"
type: "list(int)"
default_value {
list {
}
}
}
is_stateful: true
}
op {
name: "EnsureShape"
input_arg {

View File

@ -5926,6 +5926,13 @@ op {
b: false
}
}
attr {
name: "clip_boxes"
type: "bool"
default_value {
b: true
}
}
}
op {
name: "CompareAndBitpack"
@ -10897,6 +10904,14 @@ op {
name: "table_ids"
type: "list(int)"
}
attr {
name: "max_sequence_lengths"
type: "list(int)"
default_value {
list {
}
}
}
is_stateful: true
}
op {