Searched refs:grad_t (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/eager/ |
| D | function_gradients_test.py | 236 grad_t, = backprop.gradients_function(sq, [0])(t)
237 self.assertAllEqual(grad_t, [[6, 6], [14, 14]])
707 grad_t, = backprop.gradients_function(sq, [0])(t)
708 self.assertAllEqual(grad_t, [[6, 6], [14, 14]])
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_kernels.cc | 1137 const Tensor& grad_t = context->input(2); in Compute() local
1143 if (!available_tensors_.emplace(batch_key, grad_t).second) { in Compute()
1181 TensorShape output_shape(grad_t.shape()); in Compute()
|