diff options
Diffstat (limited to 'tensorflow/core/ops/nn_ops.cc')
-rw-r--r-- | tensorflow/core/ops/nn_ops.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/ops/nn_ops.cc b/tensorflow/core/ops/nn_ops.cc index f1bbfac5e6..f947d4c30d 100644 --- a/tensorflow/core/ops/nn_ops.cc +++ b/tensorflow/core/ops/nn_ops.cc @@ -432,7 +432,7 @@ REGISTER_OP("FusedResizeAndPadConv2D") .Input("paddings: int32") .Input("filter: T") .Output("output: T") - .Attr("T: {float}") + .Attr("T: {half, float, double}") .Attr("resize_align_corners: bool = false") .Attr(GetMirrorPadModeAttrString()) .Attr("strides: list(int)") @@ -446,7 +446,7 @@ REGISTER_OP("FusedPadConv2D") .Input("paddings: int32") .Input("filter: T") .Output("output: T") - .Attr("T: {float}") + .Attr("T: {half, float, double}") .Attr(GetMirrorPadModeAttrString()) .Attr("strides: list(int)") .Attr(GetPaddingAttrString()) |