Searched refs:SplitGrad (Results 1 - 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| H A D | array_grad.cc | 227 Status SplitGrad(const AttrSlice& attrs, FunctionDef* g) { function in namespace:tensorflow
242 VLOG(1) << "SplitGrad " << DebugString(*g);
245 REGISTER_OP_GRADIENT("Split", SplitGrad);
|
| H A D | array_grad_test.cc | 213 std::vector<Tensor> SplitGrad(int dim, const Tensor& x, const Tensor& dy0, function in namespace:tensorflow::__anon26715
241 TEST(ArrayGradTest, SplitGrad) {
248 auto dx = SplitGrad(1, x, dy0, dy1);
|
| /external/tensorflow/tensorflow/cc/framework/ |
| H A D | gradient_checker_test.cc | 123 TEST(GradientCheckerTest, SplitGrad) {
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| H A D | array_grad.cc | 114 Status SplitGrad(const Scope& scope, const Operation& op, function in namespace:tensorflow::ops::__anon25351
121 REGISTER_GRADIENT_OP("Split", SplitGrad);
|
| H A D | array_grad_test.cc | 101 TEST_F(ArrayGradTest, SplitGrad) {
|
Completed in 297 milliseconds