Update ops-related pbtxt files.

PiperOrigin-RevId: 312775359
Change-Id: Iffdf619b6352f62ea92362d7419c7bc16a423685
This commit is contained in:
A. Unique TensorFlower 2020-05-21 17:59:10 -07:00 committed by TensorFlower Gardener
parent 063a040287
commit c7229fcabb
5 changed files with 262 additions and 0 deletions

View File

@ -95,3 +95,59 @@ op {
}
}
}
op {
name: "SparseSegmentMean"
input_arg {
name: "data"
type_attr: "T"
}
input_arg {
name: "indices"
type_attr: "Tidx"
}
input_arg {
name: "segment_ids"
type_attr: "Tsegmentids"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
attr {
name: "Tidx"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "Tsegmentids"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
}

View File

@ -129,3 +129,76 @@ op {
}
}
}
op {
name: "SparseSegmentMeanWithNumSegments"
input_arg {
name: "data"
type_attr: "T"
}
input_arg {
name: "indices"
type_attr: "Tidx"
}
input_arg {
name: "segment_ids"
type_attr: "Tsegmentids"
}
input_arg {
name: "num_segments"
type_attr: "Tnumsegments"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
attr {
name: "Tidx"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "Tnumsegments"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "Tsegmentids"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
}

View File

@ -95,3 +95,59 @@ op {
}
}
}
op {
name: "SparseSegmentSqrtN"
input_arg {
name: "data"
type_attr: "T"
}
input_arg {
name: "indices"
type_attr: "Tidx"
}
input_arg {
name: "segment_ids"
type_attr: "Tsegmentids"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
attr {
name: "Tidx"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "Tsegmentids"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
}

View File

@ -129,3 +129,76 @@ op {
}
}
}
op {
name: "SparseSegmentSqrtNWithNumSegments"
input_arg {
name: "data"
type_attr: "T"
}
input_arg {
name: "indices"
type_attr: "Tidx"
}
input_arg {
name: "segment_ids"
type_attr: "Tsegmentids"
}
input_arg {
name: "num_segments"
type_attr: "Tnumsegments"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
attr {
name: "Tidx"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "Tnumsegments"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "Tsegmentids"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
}

View File

@ -46097,6 +46097,7 @@ op {
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_FLOAT
type: DT_DOUBLE
}
@ -46215,6 +46216,7 @@ op {
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_FLOAT
type: DT_DOUBLE
}
@ -46283,6 +46285,7 @@ op {
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_FLOAT
type: DT_DOUBLE
}
@ -46401,6 +46404,7 @@ op {
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_FLOAT
type: DT_DOUBLE
}