Searched refs:grad_t (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_relu_op_test.cc | 79 Tensor grad_t(DT_FLOAT, shape); in Activation() local
80 grad_t.flat<float>().setRandom(); in Activation()
81 Node* grad = test::graph::Constant(graph, grad_t, "grad"); in Activation()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | function_gradients_test.py | 412 grad_t, = backprop.gradients_function(sq, [0])(t)
413 self.assertAllEqual(grad_t, [[6, 6], [14, 14]])
882 grad_t, = backprop.gradients_function(sq, [0])(t)
883 self.assertAllEqual(grad_t, [[6, 6], [14, 14]])
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_kernels.cc | 807 const Tensor& grad_t = context->input(2); in Compute() local
813 if (!available_tensors_.emplace(batch_key, grad_t).second) { in Compute()
851 TensorShape output_shape(grad_t.shape()); in Compute()
|