| 121 | REGISTER_GRADIENT_OP("Split", SplitGrad); |
| 122 | |
| 123 | Status SplitVGrad(const Scope& scope, const Operation& op, |
| 124 | const std::vector<Output>& grad_inputs, |
| 125 | std::vector<Output>* grad_outputs) { |
| 126 | if (op.num_inputs() < 3) { |
| 127 | return errors::InvalidArgument("SplitV requires 3 arguments"); |
| 128 | } |
| 129 | grad_outputs->push_back(Concat(scope, grad_inputs, op.input(2))); |
| 130 | for (int i = 0; i < op.num_inputs() - 1; ++i) { |
| 131 | grad_outputs->push_back(NoGradient()); |
| 132 | } |
| 133 | return scope.status(); |
| 134 | } |
| 135 | REGISTER_GRADIENT_OP("SplitV", SplitVGrad); |
| 136 | |
| 137 | Status FillGrad(const Scope& scope, const Operation& op, |
nothing calls this directly
no test coverage detected