| 339 | REGISTER_GRADIENT_OP("AvgPool3D", AvgPool3DGradHelper); |
| 340 | |
| 341 | Status LRNGradHelper(const Scope& scope, const Operation& op, |
| 342 | const std::vector<Output>& grad_inputs, |
| 343 | std::vector<Output>* grad_outputs) { |
| 344 | auto dx = internal::LRNGrad(scope, grad_inputs[0], op.input(0), op.output(0)); |
| 345 | grad_outputs->push_back(dx); |
| 346 | return scope.status(); |
| 347 | } |
| 348 | REGISTER_GRADIENT_OP("LRN", LRNGradHelper); |
| 349 | |
| 350 | Status SoftplusGradHelper(const Scope& scope, const Operation& op, |