diff options
author | 2018-09-06 14:53:26 -0700 | |
---|---|---|
committer | 2018-09-06 15:04:52 -0700 | |
commit | e0a8285d9563122a75d94a54352f5c94f287e810 (patch) | |
tree | 4606243754011a26d4a687ecf3fbc3d9e5956bf2 /tensorflow/core/BUILD | |
parent | 7151af74292924b59b38cac7094dd64e9d38fa84 (diff) |
Simplify BUILD rule for MKL transpose op.
There is no reason for outside dependents to make a distinction between the
Eigen or MKL transpose operation, as the substitution is transparent. There is
also no need for transpose_op.cc itself to be compiled differently based on
whether MKL is in use or not. Therefore we remove external dependencies on
:mkl_transpose_op and make :transpose_op depend on it if needed (i.e., if
using MKL). This is consistent with how other transparent MKL operations (e.g.
matmul) are built.
PiperOrigin-RevId: 211874336
Diffstat (limited to 'tensorflow/core/BUILD')
-rw-r--r-- | tensorflow/core/BUILD | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/BUILD b/tensorflow/core/BUILD index f74379fca5..38eb49760c 100644 --- a/tensorflow/core/BUILD +++ b/tensorflow/core/BUILD @@ -1354,6 +1354,7 @@ cc_library( "//tensorflow/core/kernels:mkl_relu_op", "//tensorflow/core/kernels:mkl_reshape_op", "//tensorflow/core/kernels:mkl_softmax_op", + "//tensorflow/core/kernels:mkl_transpose_op", "//tensorflow/core/kernels:mkl_tfconv_op", "//tensorflow/core/kernels:mkl_aggregate_ops", ]) + if_cuda([ |