Searched refs:ReverseGrad (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_grad.cc | 397 Status ReverseGrad(const AttrSlice& attrs, FunctionDef* g) { in ReverseGrad() function
413 REGISTER_OP_GRADIENT("Reverse", ReverseGrad);
|
| D | array_grad_test.cc | 452 std::vector<Tensor> ReverseGrad(const Tensor& x, const Tensor& dims, in ReverseGrad() function
474 TEST(ArrayGradTest, ReverseGrad) { in TEST() argument
480 auto dx = ReverseGrad(x, dims, dy); in TEST()
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 258 Status ReverseGrad(const Scope& scope, const Operation& op, in ReverseGrad() function
266 REGISTER_GRADIENT_OP("ReverseV2", ReverseGrad);
|
| D | array_grad_test.cc | 217 TEST_F(ArrayGradTest, ReverseGrad) { in TEST_F() argument
|