Update ops-related pbtxt files.

PiperOrigin-RevId: 327270544
Change-Id: I16d034b0c664d6bd57d1fe3813a86b6d81ea9f66
This commit is contained in:
A. Unique TensorFlower 2020-08-18 11:30:54 -07:00 committed by TensorFlower Gardener
parent b7dbaf4f23
commit 4dc7c6aa3e
2 changed files with 100 additions and 0 deletions

View File

@ -0,0 +1,50 @@
op {
name: "IsotonicRegression"
input_arg {
name: "input"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "output_dtype"
}
output_arg {
name: "segments"
type: DT_INT32
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_INT64
type: DT_BFLOAT16
type: DT_UINT16
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "output_dtype"
type: "type"
default_value {
type: DT_FLOAT
}
allowed_values {
list {
type: DT_HALF
type: DT_BFLOAT16
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}

View File

@ -19863,6 +19863,56 @@ op {
}
allows_uninitialized_input: true
}
op {
name: "IsotonicRegression"
input_arg {
name: "input"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "output_dtype"
}
output_arg {
name: "segments"
type: DT_INT32
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_INT64
type: DT_BFLOAT16
type: DT_UINT16
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
attr {
name: "output_dtype"
type: "type"
default_value {
type: DT_FLOAT
}
allowed_values {
list {
type: DT_HALF
type: DT_BFLOAT16
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}
op {
name: "Iterator"
output_arg {