OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-07 00:20:20 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Fast path for tf.conj when it should be pass-through.

Browse Source 
PiperOrigin-RevId: 171053662
This commit is contained in:
Alexandre Passos 2017-10-04 13:26:11 -07:00 committed by TensorFlower Gardener
parent c41dbc3c18
commit cc8ee6c0f5
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/python/ops/math_ops.py
View File

@ -2317,6 +2317,10 @@ def conj(x, name=None):
Raises: Raises:
TypeError: If `x` is not a numeric tensor. TypeError: If `x` is not a numeric tensor.
""" """
if isinstance(x, ops.Tensor):
dt = x.dtype
if dt.is_floating or dt.is_integer:
return x
with ops.name_scope(name, "Conj", [x]) as name: with ops.name_scope(name, "Conj", [x]) as name:
x = ops.convert_to_tensor(x, name="x") x = ops.convert_to_tensor(x, name="x")
if x.dtype.is_complex or x.dtype == dtypes.variant: if x.dtype.is_complex or x.dtype == dtypes.variant: