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

Merge pull request #37518 from ROCmSoftwarePlatform:google_upstream_rocm_csb_fix_200311 

PiperOrigin-RevId: 304743550
Change-Id: I0c14e5aec2f6f1cbc496f2bad0d6bd7a718c9241
This commit is contained in:
TensorFlower Gardener 2020-04-03 20:55:42 -07:00
parent b9c1faf82d 3d3bf0031a
commit e530f61b9e
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/core/BUILD
View File

@ -68,6 +68,7 @@ load(
"cc_header_only_library",
"if_android",
"if_chromiumos",
"if_cuda_or_rocm",
"if_ios",
"if_mobile",
"if_not_windows",
@ -1076,8 +1077,9 @@ cc_library(
"//tensorflow/core/kernels:mkl_matmul_op",
"//tensorflow/core/kernels:mkl_tfconv_op",
"//tensorflow/core/kernels:mkl_tmp_bf16_ops",
]) + if_cuda([
]) + if_cuda_or_rocm([
"//tensorflow/core/kernels:cudnn_rnn_kernels",
]) + if_cuda([
"//tensorflow/core/grappler/optimizers:gpu_swapping_kernels",
"//tensorflow/core/grappler/optimizers:gpu_swapping_ops",
]) + if_nccl([