| 177 | REGISTER_GRADIENT_OP("MatrixDiag", MatrixDiagGrad); |
| 178 | |
| 179 | Status MatrixBandPartGrad(const Scope& scope, const Operation& op, |
| 180 | const std::vector<Output>& grad_inputs, |
| 181 | std::vector<Output>* grad_outputs) { |
| 182 | auto num_lower = op.input(1); |
| 183 | auto num_upper = op.input(2); |
| 184 | grad_outputs->push_back( |
| 185 | MatrixBandPart(scope, grad_inputs[0], num_lower, num_upper)); |
| 186 | grad_outputs->push_back(NoGradient()); |
| 187 | grad_outputs->push_back(NoGradient()); |
| 188 | return scope.status(); |
| 189 | } |
| 190 | REGISTER_GRADIENT_OP("MatrixBandPart", MatrixBandPartGrad); |
| 191 | |
| 192 | Status GatherNdGrad(const Scope& scope, const Operation& op, |
nothing calls this directly
no test coverage detected