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
|
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 {
|
op {
|
||||||
name: "Iterator"
|
name: "Iterator"
|
||||||
output_arg {
|
output_arg {
|
||||||
|
Loading…
x
Reference in New Issue
Block a user