From 85d40e74aca5ed0d8aa432c8279a301b422320a4 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Wed, 22 Apr 2020 10:59:43 -0700 Subject: [PATCH] Internal change PiperOrigin-RevId: 307852296 Change-Id: Iac01c33449b93c3e71dc2461733c73ec834d8826 --- tensorflow/python/tpu/bfloat16.py | 2 +- tensorflow/python/tpu/bfloat16_test.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/python/tpu/bfloat16.py b/tensorflow/python/tpu/bfloat16.py index b4ab1f8731f..70f71815e51 100644 --- a/tensorflow/python/tpu/bfloat16.py +++ b/tensorflow/python/tpu/bfloat16.py @@ -76,5 +76,5 @@ def bfloat16_scope(): 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 diff --git a/tensorflow/python/tpu/bfloat16_test.py b/tensorflow/python/tpu/bfloat16_test.py index be7b014a3af..78157ea86c2 100644 --- a/tensorflow/python/tpu/bfloat16_test.py +++ b/tensorflow/python/tpu/bfloat16_test.py @@ -31,7 +31,7 @@ class BFloat16ScopeTest(test.TestCase): def testScopeName(self): """Test if name for the variable scope is propagated correctly.""" with bfloat16.bfloat16_scope() as bf: - self.assertEqual(bf.name, "bfloat16") + self.assertEqual(bf.name, "") @test_util.run_deprecated_v1 def testRequestedDType(self):