Searched refs:UnpackGrad (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_grad.cc | 118 Status UnpackGrad(const AttrSlice& attrs, FunctionDef* g) { in UnpackGrad() function
140 REGISTER_OP_GRADIENT("Unpack", UnpackGrad);
|
| D | array_grad_test.cc | 77 std::vector<Tensor> UnpackGrad(const Tensor& x, const Tensor& dy0, in UnpackGrad() function
104 TEST(ArrayGradTest, UnpackGrad) { in TEST() argument
111 auto dx = UnpackGrad(x, dy0, dy1, 0); in TEST()
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 59 Status UnpackGrad(const Scope& scope, const Operation& op, in UnpackGrad() function
67 REGISTER_GRADIENT_OP("Unpack", UnpackGrad);
|