Searched refs:ReshapeGrad (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_grad.cc | 33 Status ReshapeGrad(const AttrSlice& attrs, FunctionDef* g) { in ReshapeGrad() function
51 REGISTER_OP_GRADIENT("Reshape", ReshapeGrad);
52 REGISTER_OP_GRADIENT("ExpandDims", ReshapeGrad);
|
| D | array_grad_test.cc | 307 std::vector<Tensor> ReshapeGrad(const Tensor& x, const Tensor& s, in ReshapeGrad() function
329 TEST(ArrayGradTest, ReshapeGrad) { in TEST() argument
335 auto dx = ReshapeGrad(x, s, dy); in TEST()
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 203 Status ReshapeGrad(const Scope& scope, const Operation& op, in ReshapeGrad() function
211 REGISTER_GRADIENT_OP("Reshape", ReshapeGrad);
|
| D | array_grad_test.cc | 176 TEST_F(ArrayGradTest, ReshapeGrad) { in TEST_F() argument
|