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

Merge pull request #41481 from reedwm:tf32_actual_fix

Browse Source 
PiperOrigin-RevId: 322203289
Change-Id: I87014508e2ced293049f784caa0e3905fc6fd03b
This commit is contained in:
TensorFlower Gardener 2020-07-20 12:47:13 -07:00
commit 67f914e794
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/core/common_runtime/gpu/BUILD
View File

@ -159,6 +159,7 @@ tf_cuda_library(
"//tensorflow/core:lib_internal",
"//tensorflow/core:protos_all_cc",
"//tensorflow/core:stream_executor",
"//tensorflow/core/platform:tf32_utils",
"//tensorflow/core/profiler/lib:annotated_traceme",
"//tensorflow/core/profiler/lib:scoped_annotation",
"//third_party/eigen3",

5
tensorflow/python/BUILD
View File

@ -6088,7 +6088,6 @@ pywrap_tensorflow_macro(
"//tensorflow/lite/delegates/flex:delegate",
"//tensorflow/core/profiler/internal/cpu:python_tracer",
"//tensorflow/tools/graph_transforms:transform_graph_lib",
"//tensorflow/core/platform:tf32_utils",
"//tensorflow/lite/toco/python:toco_python_api",
"//tensorflow/python/eager:pywrap_tfe_lib",
"//tensorflow/core/util/tensor_bundle",
@ -6100,7 +6099,9 @@ pywrap_tensorflow_macro(
"@ngraph_tf//:ngraph_tf",
]) + if_xla_available([
"//tensorflow/compiler/aot:tfcompile_lib",
]) + if_tpu(["//tensorflow/core/tpu:tpu_api_dlsym_initializer"]),
]) + if_tpu([
"//tensorflow/core/tpu:tpu_api_dlsym_initializer",
]) + if_static(extra_deps = ["//tensorflow/core/platform:tf32_utils"]),
)
# ** Targets for Windows build (start) **