Addressing review comments
This commit is contained in:
parent
e3fa679192
commit
55daab6ac4
@ -7771,12 +7771,7 @@ tf_mkl_kernel_library(
|
||||
hdrs = ["no_op.h"],
|
||||
prefix = "mkl_tmp_bf16_ops",
|
||||
deps = [
|
||||
":bounds_check",
|
||||
":ops_util",
|
||||
"//tensorflow/core:core_cpu",
|
||||
"//tensorflow/core:framework",
|
||||
"//tensorflow/core:lib",
|
||||
"//tensorflow/core:lib_internal",
|
||||
":no_op",
|
||||
] + mkl_deps(),
|
||||
)
|
||||
|
||||
|
@ -56,6 +56,7 @@ namespace tensorflow {
|
||||
|
||||
TF_CALL_bfloat16(REGISTER_CPU);
|
||||
#undef REGISTER_CPU
|
||||
|
||||
}
|
||||
|
||||
#endif // ENABLE_INTEL_MKL_BFLOAT16
|
||||
|
Loading…
Reference in New Issue
Block a user