Update ops-related pbtxt files.
PiperOrigin-RevId: 327270544 Change-Id: I16d034b0c664d6bd57d1fe3813a86b6d81ea9f66
This commit is contained in:
parent
b7dbaf4f23
commit
4dc7c6aa3e
@ -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
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
@ -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 {
|
||||
|
Loading…
x
Reference in New Issue
Block a user