This enables variables to be read as bfloat16 type when using get_variable.
"""
with variable_scope.variable_scope(
- 'bfloat16', custom_getter=_get_custom_getter()) as varscope:
+ '', custom_getter=_get_custom_getter()) as varscope:
yield varscope
"""Test if name for the variable scope is propogated correctly.
"""
with bfloat16.bfloat16_scope() as bf:
- self.assertEqual(bf.name, "bfloat16")
+ self.assertEqual(bf.name, "")
def testRequestedDType(self):
"""Test if requested dtype is honored in the getter.