message: [TF] Add missing headers to mkldnn build to fix build when TF is built as a submodule.
  Fixes JAX build.
diff: @@ -47,8 +47,11 @@ cc_library(
  "src/cpu/gemm/*.cpp",
  "src/cpu/gemm/*.hpp",
  "src/cpu/gemm/f32/*.cpp",
  + "src/cpu/gemm/f32/*.hpp",
  "src/cpu/gemm/s8x8s32/*.cpp",
  + "src/cpu/gemm/s8x8s32/*.hpp",
  "src/cpu/rnn/*.cpp",
  + "src/cpu/rnn/*.hpp",
  "src/cpu/xbyak/*.h",
  ]) + [":mkldnn_version_h"],
  hdrs = glob(["include/*"]),
  @@ -105,8 +108,11 @@ cc_library(
  "src/cpu/gemm/*.cpp",
  "src/cpu/gemm/*.hpp",
  "src/cpu/gemm/f32/*.cpp",
  + "src/cpu/gemm/f32/*.hpp",
  "src/cpu/gemm/s8x8s32/*.cpp",
  + "src/cpu/gemm/s8x8s32/*.hpp",
  "src/cpu/rnn/*.cpp",
  + "src/cpu/rnn/*.hpp",
  "src/cpu/xbyak/*.h",
  ]) + [":mkldnn_version_h"],
  hdrs = glob(["include/*"]),
  