Home
last modified time | relevance | path

Searched refs:PackGrad (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Darray_grad.cc92 Status PackGrad(const AttrSlice& attrs, FunctionDef* g) { in PackGrad() function
116 REGISTER_OP_GRADIENT("Pack", PackGrad);
Darray_grad_test.cc36 std::vector<Tensor> PackGrad(const Tensor& x0, const Tensor& x1, in PackGrad() function
63 TEST(ArrayGradTest, PackGrad) { in TEST() argument
70 auto dx = PackGrad(x0, x1, dy, 0); in TEST()
/external/tensorflow/tensorflow/cc/gradients/
Darray_grad.cc42 Status PackGrad(const Scope& scope, const Operation& op, in PackGrad() function
57 REGISTER_GRADIENT_OP("Pack", PackGrad);