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

Addressing review comments - round 2

Browse Source
This commit is contained in:
Niranjan Hasabnis 2019-10-14 19:02:45 -07:00
parent 55daab6ac4
commit 79e09071d9
2 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/core/kernels/BUILD
View File

@ -7768,7 +7768,6 @@ tf_mkl_kernel_library(
tf_mkl_kernel_library(
name = "mkl_tmp_bf16_ops",
hdrs = ["no_op.h"],
prefix = "mkl_tmp_bf16_ops",
deps = [
":no_op",

2
tensorflow/core/kernels/cwise_op_reciprocal.cc
View File

@ -38,7 +38,7 @@ REGISTER6(UnaryOp, CPU, "Reciprocal", functor::inverse, float, Eigen::half,
#else
REGISTER5(UnaryOp, CPU, "Reciprocal", functor::inverse, float, Eigen::half,
double, complex64, complex128);
#endif
#endif // ENABLE_INTEL_MKL_BFLOAT16
#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
REGISTER4(UnaryOp, GPU, "Reciprocal", functor::inverse, float, Eigen::half,
double, int64);