| 597 | REGISTER_GRADIENT_OP("Imag", ImagGrad); |
| 598 | |
| 599 | Status ComplexGrad(const Scope& scope, const Operation& op, |
| 600 | const std::vector<Output>& grad_inputs, |
| 601 | std::vector<Output>* grad_outputs) { |
| 602 | auto gx_1 = Real(scope, grad_inputs[0]); |
| 603 | auto gx_2 = Imag(scope, grad_inputs[0]); |
| 604 | return BinaryGradCommon(scope, op, grad_outputs, gx_1, gx_2); |
| 605 | } |
| 606 | REGISTER_GRADIENT_OP("Complex", ComplexGrad); |
| 607 | |
| 608 | Status AngleGrad(const Scope& scope, const Operation& op, |
nothing calls this directly
no test coverage detected