[INTEL MKL] Changed the default value of input parameter to None to align with API policy
This commit is contained in:
parent
5ee6785046
commit
f484a66802
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user