diff options
author | 2016-05-23 11:39:39 -0800 | |
---|---|---|
committer | 2016-05-23 12:42:36 -0700 | |
commit | 892ca4ddc12852a7b4633fd08f163941356cb4e6 (patch) | |
tree | be913f46bb9323685c5a807a89fca6dc52a25504 /tensorflow/core/ops/math_grad.cc | |
parent | 76d90938f95a14a5723c253ec8529e93939a25e2 (diff) |
Merge changes from github.
Change: 123026122
Diffstat (limited to 'tensorflow/core/ops/math_grad.cc')
-rw-r--r-- | tensorflow/core/ops/math_grad.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/ops/math_grad.cc b/tensorflow/core/ops/math_grad.cc index d290580077..7b4b28159d 100644 --- a/tensorflow/core/ops/math_grad.cc +++ b/tensorflow/core/ops/math_grad.cc @@ -284,7 +284,7 @@ REGISTER_OP_GRADIENT("Sub", SubGrad); Status MulGrad(const AttrSlice& attrs, FunctionDef* g) { DataType T; TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "T", &T)); - if (T == DT_COMPLEX64) { + if (T == DT_COMPLEX64 || T == DT_COMPLEX128) { return GradForBinaryCwise( g, { {{"cy"}, "Conj", {"y"}, {}, {"dz"}}, @@ -543,7 +543,7 @@ Status MatMulGradCommon(const string& opname, const string& attr_adj_x, FunctionDef* g) { DataType T; TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "T", &T)); - if (T == DT_COMPLEX64) { + if (T == DT_COMPLEX64 || T == DT_COMPLEX128) { return errors::Unimplemented( "MatMul gradient for complex is not supported yet."); } |