Searched refs:in_grad (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients_util.py | 707 for i, (t_in, in_grad) in enumerate(zip(_Inputs(op, xs_set), in_grads)):
708 if in_grad is not None:
709 if (isinstance(in_grad, ops.Tensor) and
712 in_grad.set_shape(t_in.get_shape())
719 (op.name, i, t_in.shape, in_grad.shape))
721 _SetGrad(grads, t_in, in_grad)
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | gradients_test.cc | 534 std::vector<Tensor> in_grad; in TEST_F() local
535 TF_ASSERT_OK(session.Run(grad_outputs, &in_grad)); in TEST_F()
537 test::ExpectTensorEqual<int>(in_grad[i], test::AsTensor<int>({i}, {1})); in TEST_F()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | tape.h | 1007 Gradient* in_grad = in_grads[grad_index]; in ForwardpropFromTape() local
1008 if (in_grad != nullptr) { in ForwardpropFromTape()
1010 vspace_.MarkAsResult(in_grad); in ForwardpropFromTape()
1012 used_in_grads.push_back(in_grad); in ForwardpropFromTape()
|