Update ops-related pbtxt files.
PiperOrigin-RevId: 350272682 Change-Id: I7e1d33caa1ef4095e8dccb71ae2b769cc94962c8
This commit is contained in:
parent
d57f956517
commit
51463f4b40
@ -77,3 +77,47 @@ op {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
op {
|
||||||
|
name: "ResizeNearestNeighborGrad"
|
||||||
|
input_arg {
|
||||||
|
name: "grads"
|
||||||
|
type_attr: "T"
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "size"
|
||||||
|
type: DT_INT32
|
||||||
|
}
|
||||||
|
output_arg {
|
||||||
|
name: "output"
|
||||||
|
type_attr: "T"
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "T"
|
||||||
|
type: "type"
|
||||||
|
allowed_values {
|
||||||
|
list {
|
||||||
|
type: DT_UINT8
|
||||||
|
type: DT_INT8
|
||||||
|
type: DT_INT32
|
||||||
|
type: DT_HALF
|
||||||
|
type: DT_FLOAT
|
||||||
|
type: DT_DOUBLE
|
||||||
|
type: DT_BFLOAT16
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "align_corners"
|
||||||
|
type: "bool"
|
||||||
|
default_value {
|
||||||
|
b: false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "half_pixel_centers"
|
||||||
|
type: "bool"
|
||||||
|
default_value {
|
||||||
|
b: false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
@ -37307,6 +37307,7 @@ op {
|
|||||||
type: DT_HALF
|
type: DT_HALF
|
||||||
type: DT_FLOAT
|
type: DT_FLOAT
|
||||||
type: DT_DOUBLE
|
type: DT_DOUBLE
|
||||||
|
type: DT_BFLOAT16
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user