| 348 | REGISTER_GRADIENT_OP("SpaceToBatchND", SpaceToBatchNDGrad); |
| 349 | |
| 350 | Status BatchToSpaceGrad(const Scope& scope, const Operation& op, |
| 351 | const std::vector<Output>& grad_inputs, |
| 352 | std::vector<Output>* grad_outputs) { |
| 353 | int block_size; |
| 354 | TF_RETURN_IF_ERROR( |
| 355 | GetNodeAttr(op.node()->attrs(), "block_size", &block_size)); |
| 356 | grad_outputs->push_back( |
| 357 | SpaceToBatch(scope, grad_inputs[0], op.input(1), block_size)); |
| 358 | grad_outputs->push_back(NoGradient()); |
| 359 | return scope.status(); |
| 360 | } |
| 361 | REGISTER_GRADIENT_OP("BatchToSpace", BatchToSpaceGrad); |
| 362 | |
| 363 | Status BatchToSpaceNDGrad(const Scope& scope, const Operation& op, |
nothing calls this directly
no test coverage detected