Searched refs:SplitVGrad (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_grad.cc | 247 Status SplitVGrad(const AttrSlice& attrs, FunctionDef* g) { in SplitVGrad() function
266 REGISTER_OP_GRADIENT("SplitV", SplitVGrad);
|
| D | array_grad_test.cc | 241 std::vector<Tensor> SplitVGrad(const Tensor& x, const Tensor& size_splits, in SplitVGrad() function
300 auto dx = SplitVGrad(x, size_splits, 1, dy0, dy1); in TEST()
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 132 Status SplitVGrad(const Scope& scope, const Operation& op, in SplitVGrad() function
144 REGISTER_GRADIENT_OP("SplitV", SplitVGrad);
|
| D | array_grad_test.cc | 111 TEST_F(ArrayGradTest, SplitVGrad) { in TEST_F() argument
|