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

tf.numpy avoid using typing module given Kokoro breakage.

Browse Source 
PiperOrigin-RevId: 317396321
Change-Id: Ifa05dcbbdd8998708e52ba52cdb03e33e4952f21
This commit is contained in:
A. Unique TensorFlower 2020-06-19 15:58:05 -07:00 committed by TensorFlower Gardener
parent 1823f87735
commit 866eb4828a
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/python/ops/numpy_ops/np_array_ops.py
View File

@ -21,7 +21,6 @@ from __future__ import print_function
import math import math
import numbers import numbers
from typing import Sequence
import numpy as np import numpy as np
import six import six
@ -1684,10 +1683,10 @@ def _slice_helper(tensor, slice_spec):
def _as_spec_tuple(slice_spec): def _as_spec_tuple(slice_spec):
"""Convert slice_spec to tuple.""" """Convert slice_spec to tuple."""
if isinstance(slice_spec, if isinstance(slice_spec,
Sequence) and not isinstance(slice_spec, np.ndarray): (list, tuple)) and not isinstance(slice_spec, np.ndarray):
is_index = True is_index = True
for s in slice_spec: for s in slice_spec:
if s is None or s is Ellipsis or isinstance(s, (Sequence, slice)): if s is None or s is Ellipsis or isinstance(s, (list, tuple, slice)):
is_index = False is_index = False
break break
elif isinstance(s, (np_arrays.ndarray, np.ndarray)) and s.ndim != 0: elif isinstance(s, (np_arrays.ndarray, np.ndarray)) and s.ndim != 0: