Update ops-related pbtxt files.
PiperOrigin-RevId: 261365835
This commit is contained in:
parent
b9964d43a3
commit
c11b8e458b
@ -78,3 +78,31 @@ op {
|
||||
type: "string"
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "CheckNumerics"
|
||||
input_arg {
|
||||
name: "tensor"
|
||||
type_attr: "T"
|
||||
}
|
||||
output_arg {
|
||||
name: "output"
|
||||
type_attr: "T"
|
||||
}
|
||||
attr {
|
||||
name: "T"
|
||||
type: "type"
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_BFLOAT16
|
||||
type: DT_HALF
|
||||
type: DT_FLOAT
|
||||
type: DT_DOUBLE
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "message"
|
||||
type: "string"
|
||||
}
|
||||
is_stateful: true
|
||||
}
|
||||
|
@ -6132,6 +6132,7 @@ op {
|
||||
name: "message"
|
||||
type: "string"
|
||||
}
|
||||
is_stateful: true
|
||||
}
|
||||
op {
|
||||
name: "Cholesky"
|
||||
|
Loading…
Reference in New Issue
Block a user