diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-07 19:18:02 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-07 19:22:15 -0700 |
commit | e970a022ef6a3602dd5c9ea15afa96a2291880b1 (patch) | |
tree | 11c5c0cb723c55214b451a4cbaf643aed44473d7 | |
parent | 4fd48f57cd1dcd960bea1757e1c59032db66b3d0 (diff) |
Update ops-related pbtxt files.
PiperOrigin-RevId: 212073366
-rw-r--r-- | tensorflow/core/ops/compat/ops_history.v1.pbtxt | 127 | ||||
-rw-r--r-- | tensorflow/core/ops/ops.pbtxt | 51 |
2 files changed, 172 insertions, 6 deletions
diff --git a/tensorflow/core/ops/compat/ops_history.v1.pbtxt b/tensorflow/core/ops/compat/ops_history.v1.pbtxt index c32d6f84f5..34e6b5560b 100644 --- a/tensorflow/core/ops/compat/ops_history.v1.pbtxt +++ b/tensorflow/core/ops/compat/ops_history.v1.pbtxt @@ -35790,6 +35790,42 @@ op { } } op { + name: "NonMaxSuppressionV2" + input_arg { + name: "boxes" + type_attr: "T" + } + input_arg { + name: "scores" + type_attr: "T" + } + input_arg { + name: "max_output_size" + type: DT_INT32 + } + input_arg { + name: "iou_threshold" + type: DT_FLOAT + } + output_arg { + name: "selected_indices" + type: DT_INT32 + } + attr { + name: "T" + type: "type" + default_value { + type: DT_FLOAT + } + allowed_values { + list { + type: DT_HALF + type: DT_FLOAT + } + } + } +} +op { name: "NonMaxSuppressionV3" input_arg { name: "boxes" @@ -35817,6 +35853,46 @@ op { } } op { + name: "NonMaxSuppressionV3" + input_arg { + name: "boxes" + type_attr: "T" + } + input_arg { + name: "scores" + type_attr: "T" + } + input_arg { + name: "max_output_size" + type: DT_INT32 + } + input_arg { + name: "iou_threshold" + type: DT_FLOAT + } + input_arg { + name: "score_threshold" + type: DT_FLOAT + } + output_arg { + name: "selected_indices" + type: DT_INT32 + } + attr { + name: "T" + type: "type" + default_value { + type: DT_FLOAT + } + allowed_values { + list { + type: DT_HALF + type: DT_FLOAT + } + } + } +} +op { name: "NonMaxSuppressionV4" input_arg { name: "boxes" @@ -35855,6 +35931,57 @@ op { } } op { + name: "NonMaxSuppressionV4" + input_arg { + name: "boxes" + type_attr: "T" + } + input_arg { + name: "scores" + type_attr: "T" + } + input_arg { + name: "max_output_size" + type: DT_INT32 + } + input_arg { + name: "iou_threshold" + type: DT_FLOAT + } + input_arg { + name: "score_threshold" + type: DT_FLOAT + } + output_arg { + name: "selected_indices" + type: DT_INT32 + } + output_arg { + name: "valid_outputs" + type: DT_INT32 + } + attr { + name: "T" + type: "type" + default_value { + type: DT_FLOAT + } + allowed_values { + list { + type: DT_HALF + type: DT_FLOAT + } + } + } + attr { + name: "pad_to_max_output_size" + type: "bool" + default_value { + b: false + } + } +} +op { name: "NonMaxSuppressionWithOverlaps" input_arg { name: "overlaps" diff --git a/tensorflow/core/ops/ops.pbtxt b/tensorflow/core/ops/ops.pbtxt index aeb03c5952..c00c0030e6 100644 --- a/tensorflow/core/ops/ops.pbtxt +++ b/tensorflow/core/ops/ops.pbtxt @@ -17098,11 +17098,11 @@ op { name: "NonMaxSuppressionV2" input_arg { name: "boxes" - type: DT_FLOAT + type_attr: "T" } input_arg { name: "scores" - type: DT_FLOAT + type_attr: "T" } input_arg { name: "max_output_size" @@ -17116,16 +17116,29 @@ op { name: "selected_indices" type: DT_INT32 } + attr { + name: "T" + type: "type" + default_value { + type: DT_FLOAT + } + allowed_values { + list { + type: DT_HALF + type: DT_FLOAT + } + } + } } op { name: "NonMaxSuppressionV3" input_arg { name: "boxes" - type: DT_FLOAT + type_attr: "T" } input_arg { name: "scores" - type: DT_FLOAT + type_attr: "T" } input_arg { name: "max_output_size" @@ -17143,16 +17156,29 @@ op { name: "selected_indices" type: DT_INT32 } + attr { + name: "T" + type: "type" + default_value { + type: DT_FLOAT + } + allowed_values { + list { + type: DT_HALF + type: DT_FLOAT + } + } + } } op { name: "NonMaxSuppressionV4" input_arg { name: "boxes" - type: DT_FLOAT + type_attr: "T" } input_arg { name: "scores" - type: DT_FLOAT + type_attr: "T" } input_arg { name: "max_output_size" @@ -17175,6 +17201,19 @@ op { type: DT_INT32 } attr { + name: "T" + type: "type" + default_value { + type: DT_FLOAT + } + allowed_values { + list { + type: DT_HALF + type: DT_FLOAT + } + } + } + attr { name: "pad_to_max_output_size" type: "bool" default_value { |