diff --git a/tensorflow/compiler/mlir/lite/ir/tfl_ops.td b/tensorflow/compiler/mlir/lite/ir/tfl_ops.td index af7f0294589..4e3c1c0c5f4 100644 --- a/tensorflow/compiler/mlir/lite/ir/tfl_ops.td +++ b/tensorflow/compiler/mlir/lite/ir/tfl_ops.td @@ -450,7 +450,7 @@ retained with length 1. } def TFL_TransposeConvOp: - TFL_Op<"transpose_conv", [NoSideEffect]> { + TFL_Op<"transpose_conv", [NoSideEffect, TFL_GpuTargetOp]> { let summary = "Transpose convolution operator"; let description = [{