Internal change
PiperOrigin-RevId: 307852296 Change-Id: Iac01c33449b93c3e71dc2461733c73ec834d8826
This commit is contained in:
parent
114b8ef31a
commit
85d40e74ac
@ -76,5 +76,5 @@ def bfloat16_scope():
|
|||||||
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.
|
||||||
"""
|
"""
|
||||||
with variable_scope.variable_scope(
|
with variable_scope.variable_scope(
|
||||||
'bfloat16', custom_getter=_get_custom_getter()) as varscope:
|
'', custom_getter=_get_custom_getter()) as varscope:
|
||||||
yield varscope
|
yield varscope
|
||||||
|
@ -31,7 +31,7 @@ class BFloat16ScopeTest(test.TestCase):
|
|||||||
def testScopeName(self):
|
def testScopeName(self):
|
||||||
"""Test if name for the variable scope is propagated correctly."""
|
"""Test if name for the variable scope is propagated correctly."""
|
||||||
with bfloat16.bfloat16_scope() as bf:
|
with bfloat16.bfloat16_scope() as bf:
|
||||||
self.assertEqual(bf.name, "bfloat16")
|
self.assertEqual(bf.name, "")
|
||||||
|
|
||||||
@test_util.run_deprecated_v1
|
@test_util.run_deprecated_v1
|
||||||
def testRequestedDType(self):
|
def testRequestedDType(self):
|
||||||
|
Loading…
Reference in New Issue
Block a user