Searched refs:grad_wrt_input (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_grad_test.py | 98 grad_wrt_input = gradients_impl.gradients(out, x)[0]
99 self.run_test(f, grad_wrt_input)
153 grad_wrt_input = gradients_impl.gradients(out, x)[0]
154 self.run_test(f, grad_wrt_input)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fake_quant_ops.cc | 237 Tensor* grad_wrt_input; in Compute() local
239 context->allocate_output(0, input.shape(), &grad_wrt_input)); in Compute()
253 quant_min_, quant_max_, grad_wrt_input->flat<float>(), in Compute()
381 Tensor* grad_wrt_input; in Compute() local
383 context->allocate_output(0, input.shape(), &grad_wrt_input)); in Compute()
398 quant_min_, quant_max_, grad_wrt_input->flat_inner_dims<float, 2>(), in Compute()
|