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

Add replay_computation target for mlir_gpu backend.

Browse Source 
PiperOrigin-RevId: 283500767
Change-Id: I2839e5e698427a9df1997aed2e57d5b865bfb7e6
This commit is contained in:
Adrian Kuegel 2019-12-03 01:30:07 -08:00 committed by TensorFlower Gardener
parent 12bacab7bb
commit 8fad723f04
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/compiler/xla/tools/BUILD
View File

@ -95,6 +95,14 @@ tf_cc_binary(
], ],
) )
tf_cc_binary(
name = "replay_computation_mlir_gpu",
deps = [
":replay_computation_library",
"//tensorflow/compiler/xla/service:mlir_gpu_plugin",
],
)
tf_cc_binary( tf_cc_binary(
name = "replay_computation_interpreter", name = "replay_computation_interpreter",
deps = [ deps = [