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

Addressing review comments

Browse Source
This commit is contained in:
Niranjan Hasabnis 2019-10-08 11:55:13 -07:00
parent e3fa679192
commit 55daab6ac4
2 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/core/kernels/BUILD
View File

@ -7771,12 +7771,7 @@ tf_mkl_kernel_library(
hdrs = ["no_op.h"], hdrs = ["no_op.h"],
prefix = "mkl_tmp_bf16_ops", prefix = "mkl_tmp_bf16_ops",
deps = [ deps = [
":bounds_check", ":no_op",
":ops_util",
"//tensorflow/core:core_cpu",
"//tensorflow/core:framework",
"//tensorflow/core:lib",
"//tensorflow/core:lib_internal",
] + mkl_deps(), ] + mkl_deps(),
) )

1
tensorflow/core/kernels/mkl_tmp_bf16_ops.cc
View File

@ -56,6 +56,7 @@ namespace tensorflow {
TF_CALL_bfloat16(REGISTER_CPU); TF_CALL_bfloat16(REGISTER_CPU);
#undef REGISTER_CPU #undef REGISTER_CPU
} }
#endif // ENABLE_INTEL_MKL_BFLOAT16 #endif // ENABLE_INTEL_MKL_BFLOAT16