From 8fbe04075bf13dae5a52d3676894d85fdbeca625 Mon Sep 17 00:00:00 2001
From: Renjie Liu <renjieliu@google.com>
Date: Thu, 16 Apr 2020 09:58:53 -0700
Subject: [PATCH] add transpose_conv to gpu target

PiperOrigin-RevId: 306867276
Change-Id: I4182e3fc5b0fef152f7f5b468b8484ebcd99d888
---
 tensorflow/compiler/mlir/lite/ir/tfl_ops.td | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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 = [{