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

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

Completed in 1459 milliseconds