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

[INTEL MKL] Changed the default value of input parameter to None to align with API policy

Browse Source
This commit is contained in:
Xiaoming (Jason) Cui 2020-05-27 11:10:44 -07:00
parent 5ee6785046
commit f484a66802
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/python/tpu/bfloat16.py
View File

@ -70,11 +70,13 @@ def _get_custom_getter():
@tf_export(v1=['tpu.bfloat16_scope']) @tf_export(v1=['tpu.bfloat16_scope'])
@tf_contextlib.contextmanager @tf_contextlib.contextmanager
def bfloat16_scope(name=''): def bfloat16_scope(name=None):
"""Scope class for bfloat16 variables so that the model uses custom getter. """Scope class for bfloat16 variables so that the model uses custom getter.
This enables variables to be read as bfloat16 type when using get_variable. This enables variables to be read as bfloat16 type when using get_variable.
""" """
if name is None:
name = ''
with variable_scope.variable_scope( with variable_scope.variable_scope(
name, custom_getter=_get_custom_getter()) as varscope: name, custom_getter=_get_custom_getter()) as varscope:
yield varscope yield varscope