parent
0db03690e2
commit
33820a9e52
@ -701,11 +701,6 @@ def _SqueezeGrad(op, grad):
|
|||||||
def _TransposeGrad(op, grad):
|
def _TransposeGrad(op, grad):
|
||||||
"""Returns unshuffle(grad)."""
|
"""Returns unshuffle(grad)."""
|
||||||
p = op.inputs[1]
|
p = op.inputs[1]
|
||||||
if not context.executing_eagerly():
|
|
||||||
p_static = pywrap_tensorflow.TF_TryEvaluateConstant_wrapper(
|
|
||||||
p.graph._c_graph, p._as_tf_output()) # pylint: disable=protected-access
|
|
||||||
if p_static is not None:
|
|
||||||
p = constant_op.constant(p_static, dtype=p.dtype)
|
|
||||||
return [array_ops.transpose(grad, array_ops.invert_permutation(p)), None]
|
return [array_ops.transpose(grad, array_ops.invert_permutation(p)), None]
|
||||||
|
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user