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

This commit is contained in:
Xiaoming (Jason) Cui 2020-05-27 11:10:44 -07:00
parent 5ee6785046
commit f484a66802

View File

@ -70,11 +70,13 @@ def _get_custom_getter():
@tf_export(v1=['tpu.bfloat16_scope'])
@tf_contextlib.contextmanager
def bfloat16_scope(name=''):
def bfloat16_scope(name=None):
"""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.
"""
if name is None:
name = ''
with variable_scope.variable_scope(
name, custom_getter=_get_custom_getter()) as varscope:
yield varscope