get type from self

Signed-off-by: Lyaction <[email protected]>
DeepRec-AI · Apr 12, 2024 · e919f74 · e919f74
1 parent 9091425
commit e919f74
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tensorflow/python/ops/kv_variable_ops.py b/tensorflow/python/ops/kv_variable_ops.py
@@ -743,9 +743,7 @@ def sparse_read(self, indices, name=None, ev_init_value=None, counts=None):
         default_value = ev_init_value
         is_use_default_value_tensor = True
       else:
-        from tensorflow.python.ops import variable_scope
-        default_type = variable_scope.get_variable_scope().dtype
-        default_value = ops.convert_to_tensor(1.0, dtype=default_type)
+        default_value = ops.convert_to_tensor(1.0, dtype=self.dtype)
         is_use_default_value_tensor = False
       if counts != None:
         value = gen_kv_variable_ops.kv_resource_gather_v1(self._handle,