experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Remove usage of internal sparse_tensor.is_sparse from keras

Browse Source 
PiperOrigin-RevId: 341536761
Change-Id: Iaf7c46746bd5245e15ddf963fa014aeccad300e1
This commit is contained in:
Tomer Kaftan 2020-11-09 20:02:30 -08:00 committed by TensorFlower Gardener
parent 039cb9d993
commit 0027d1b3c3
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/python/keras/engine/base_layer_utils.py
View File

@ -213,7 +213,8 @@ def _create_keras_history_helper(tensors, processed_ops, created_layers):
for tensor in tensor_list:
if getattr(tensor, '_keras_history', None) is not None:
continue
if sparse_tensor.is_sparse(tensor):
if isinstance(
tensor, (sparse_tensor.SparseTensor, sparse_tensor.SparseTensorValue)):
sparse_ops.append(tensor.op)
continue
if tf_utils.is_ragged(tensor):

2
tensorflow/python/keras/engine/base_preprocessing_layer.py
View File

@ -265,8 +265,6 @@ def convert_to_list(values, sparse_default_value=None):
values = K.get_session(values).run(values)
values = values.to_list()
# TODO(momernick): Add a sparse_tensor.is_sparse() method to replace this
# check.
if isinstance(values,
(sparse_tensor.SparseTensor, sparse_tensor.SparseTensorValue)):
if sparse_default_value is None: