experiments
/
STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add transpose_conv to gpu target 

PiperOrigin-RevId: 306867276
Change-Id: I4182e3fc5b0fef152f7f5b468b8484ebcd99d888
This commit is contained in:
Renjie Liu 2020-04-16 09:58:53 -07:00 committed by TensorFlower Gardener
parent f303d000c6
commit 8fbe04075b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
View File

@ -450,7 +450,7 @@ retained with length 1.
} }
def TFL_TransposeConvOp: def TFL_TransposeConvOp:
TFL_Op<"transpose_conv", [NoSideEffect]> { TFL_Op<"transpose_conv", [NoSideEffect, TFL_GpuTargetOp]> {
let summary = "Transpose convolution operator"; let summary = "Transpose convolution operator";
let description = [{ let description = [{