diff --git a/tensorflow/core/ops/compat/ops_history_v2/ResizeNearestNeighborGrad.pbtxt b/tensorflow/core/ops/compat/ops_history_v2/ResizeNearestNeighborGrad.pbtxt index b16307e3ba7..001525b0733 100644 --- a/tensorflow/core/ops/compat/ops_history_v2/ResizeNearestNeighborGrad.pbtxt +++ b/tensorflow/core/ops/compat/ops_history_v2/ResizeNearestNeighborGrad.pbtxt @@ -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 + } + } +} diff --git a/tensorflow/core/ops/ops.pbtxt b/tensorflow/core/ops/ops.pbtxt index ca4f14ccc0a..babff1d9bc7 100644 --- a/tensorflow/core/ops/ops.pbtxt +++ b/tensorflow/core/ops/ops.pbtxt @@ -37307,6 +37307,7 @@ op { type: DT_HALF type: DT_FLOAT type: DT_DOUBLE + type: DT_BFLOAT16 } } }