| /external/tensorflow/tensorflow/python/ops/ |
| gradients.py | 22 from tensorflow.python.ops.gradients_impl import AggregationMethod
23 from tensorflow.python.ops.gradients_impl import gradients
24 from tensorflow.python.ops.gradients_impl import hessians
|
| nn_grad_test.py | 26 from tensorflow.python.ops import gradients_impl
39 r_g = gradients_impl.gradients(r, inputs)[0]
|
| concat_benchmark.py | 30 from tensorflow.python.ops import gradients_impl
72 gradients_impl.gradients(output, inputs) for output in outputs
|
| /external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/ |
| fused_rnn_cell_test.py | 27 from tensorflow.python.ops import gradients_impl
61 basic_grads = sess.run(gradients_impl.gradients(packed_outputs, inputs))
63 gradients_impl.gradients(packed_outputs, basic_vars))
76 fused_static_grads = sess.run(gradients_impl.gradients(outputs, inputs))
78 gradients_impl.gradients(outputs, fused_static_vars))
98 gradients_impl.gradients(outputs, inputs))
100 gradients_impl.gradients(outputs, fused_dynamic_vars))
133 basic_grads = sess.run(gradients_impl.gradients(packed_outputs, inputs))
135 gradients_impl.gradients(packed_outputs, basic_vars))
154 fused_grads = sess.run(gradients_impl.gradients(outputs, inputs)
[all...] |
| gru_ops_test.py | 30 from tensorflow.python.ops import gradients_impl
178 d_new_h_wrt_x = gradients_impl.gradients([output], x)
179 d_new_h_wrt_h = gradients_impl.gradients([output], h)
180 d_new_h_wrt_w_ru = gradients_impl.gradients([output], w_ru)
181 d_new_h_wrt_w_c = gradients_impl.gradients([output], w_c)
182 d_new_h_wrt_b_ru = gradients_impl.gradients([output], b_ru)
183 d_new_h_wrt_b_c = gradients_impl.gradients([output], b_c)
199 d_new_h_wrt_x = gradients_impl.gradients([output], x)
200 d_new_h_wrt_h = gradients_impl.gradients([output], h)
201 d_new_h_wrt_w_ru = gradients_impl.gradients([output], w_ru
[all...] |
| lstm_ops_test.py | 30 from tensorflow.python.ops import gradients_impl
108 basic_grads = sess.run(gradients_impl.gradients(basic_outputs_op, inputs))
112 basic_wgrads = sess.run(gradients_impl.gradients(basic_outputs_op, xs))
115 block_grads = sess.run(gradients_impl.gradients(block_outputs_op, inputs))
116 block_wgrads = sess.run(gradients_impl.gradients(block_outputs_op, xs))
122 fused_grads = sess.run(gradients_impl.gradients(fused_outputs_op, inputs))
123 fused_wgrads = sess.run(gradients_impl.gradients(fused_outputs_op, xs))
416 gradients_impl.gradients(unfused_outputs_op, inputs))
418 gradients_impl.gradients(unfused_outputs_op, cell_vars))
422 fused_grads = sess.run(gradients_impl.gradients(fused_outputs_op, inputs)
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| gradient_correctness_test.py | 25 from tensorflow.python.ops import gradients_impl
37 grads = gradients_impl.gradients([yexp, yexplog], [x])
|
| logging_ops_test.py | 25 from tensorflow.python.ops import gradients_impl
74 wx_grad = gradients_impl.gradients(wx, w)[0]
75 wx_print_grad = gradients_impl.gradients(wx_print, w)[0]
|
| softplus_op_test.py | 25 from tensorflow.python.ops import gradients_impl
96 (grad,) = gradients_impl.gradients(y, x)
113 (grad,) = gradients_impl.gradients(y, x)
114 (grad_grad,) = gradients_impl.gradients(grad, x)
|
| relu_op_test.py | 27 from tensorflow.python.ops import gradients_impl
114 z = gradients_impl.gradients(y, x)
132 z = gradients_impl.gradients(y, x)
280 g, = gradients_impl.gradients(elu, x)
281 gg, = gradients_impl.gradients(g, x)
294 z = gradients_impl.gradients(y, x)
312 z = gradients_impl.gradients(y, x)
385 z = gradients_impl.gradients(y, x)
403 z = gradients_impl.gradients(y, x)
|
| control_flow_ops_py_test.py | 51 from tensorflow.python.ops import gradients_impl
346 grad = gradients_impl.gradients(y, [v])
643 grad = gradients_impl.gradients(r, [x])[0]
656 grad = gradients_impl.gradients(r, [x])[0]
666 result = gradients_impl.gradients(y, x)[0]
672 result = gradients_impl.gradients(z, x)[0]
683 grad = gradients_impl.gradients(r, [v1])[0]
787 g = gradients_impl.gradients(out, v)
822 gs = gradients_impl.gradients(loop_no_xla, v)
836 _ = gradients_impl.gradients(loop_no_maxiter, v
[all...] |
| dynamic_stitch_op_test.py | 27 from tensorflow.python.ops import gradients_impl
127 grads = gradients_impl.gradients(stitched_t, indices + data,
226 grads = gradients_impl.gradients(stitched_t, indices + data,
264 grads = gradients_impl.gradients(stitched_t, indices + data,
|
| ctc_loss_op_test.py | 28 from tensorflow.python.ops import gradients_impl
71 grad = gradients_impl.gradients(loss, [inputs_t])[0]
262 _ = gradients_impl._hessian_vector_product(loss, [inputs_t], v)
|
| gather_nd_op_test.py | 30 from tensorflow.python.ops import gradients_impl
226 grads = gradients_impl.gradients([outputs], [inputs], [grad_vals])[0]
237 grads = gradients_impl.gradients([outputs], [inputs], [grad_vals])[0]
252 grads = gradients_impl.gradients([outputs], [inputs], [grad_vals])[0]
278 grads = gradients_impl.gradients([outputs], [inputs], [grad_vals])[0]
296 grads = gradients_impl.gradients([outputs], [inputs], [grad_vals])[0]
314 grads = gradients_impl.gradients([outputs], [inputs], [grad_vals])[0]
|
| /external/tensorflow/tensorflow/contrib/kfac/python/ops/ |
| curvature_matrix_vector_products.py | 22 from tensorflow.python.ops import gradients_impl
91 return gradients_impl.gradients(
152 return gradients_impl.gradients(
153 gradients_impl.gradients(self._total_loss, self._wrt_tensors),
|
| estimator.py | 29 from tensorflow.python.ops import gradients_impl
285 grads_flat = gradients_impl.gradients(
293 grads_flat = gradients_impl.gradients(
311 grads_flat = gradients_impl.gradients(
327 grads_flat = gradients_impl.gradients(
|
| /external/tensorflow/tensorflow/contrib/bayesflow/python/kernel_tests/ |
| custom_grad_test.py | 27 from tensorflow.python.ops import gradients_impl
48 gx = gradients_impl.gradients(fx, x)[0]
62 gx = gradients_impl.gradients(fx, x)[0]
87 gz = gradients_impl.gradients(fz, variables.trainable_variables())
114 gx = gradients_impl.gradients(fx, variables.trainable_variables())
148 gx = gradients_impl.gradients(fx, variables.trainable_variables())
|
| monte_carlo_test.py | 29 from tensorflow.python.ops import gradients_impl
193 gradients_impl.gradients(efx_true, x)[0],
194 gradients_impl.gradients(efx_reparam, x)[0],
195 gradients_impl.gradients(efx_score, x)[0],
239 gradp = lambda fp: gradients_impl.gradients(fp, mu_p)[0]
240 gradq = lambda fq: gradients_impl.gradients(fq, mu_q)[0]
282 gradp = lambda fp: gradients_impl.gradients(fp, concentration_p)[0]
283 gradq = lambda fq: gradients_impl.gradients(fq, concentration_q)[0]
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| rev_block_lib_test.py | 28 from tensorflow.python.ops import gradients_impl
149 grads_rev = gradients_impl.gradients(loss_rev, wrt)
150 grads = gradients_impl.gradients(loss, wrt)
222 _ = gradients_impl.gradients(loss,
268 grad1 = gradients_impl.gradients(out1, recompute_vars)
269 grad2 = gradients_impl.gradients(out2, reg_vars)
296 grad_inputs = gradients_impl.gradients(
298 grad_vars = gradients_impl.gradients(
316 grads = gradients_impl.gradients(
318 custom_grads = gradients_impl.gradients
[all...] |
| rev_block_lib.py | 40 from tensorflow.python.ops import gradients_impl
97 grad_gy1_y2 = gradients_impl.gradients(gy1, y1_stop, grad_y2)[0]
100 gradients_impl.gradients(fx2, x2_stop, grad_y1)[0] + grad_y2 +
101 gradients_impl.gradients(fx2, x2_stop, grad_gy1_y2)[0])
104 grads1 = gradients_impl.gradients(gy1, g_vars + g_side_input, grad_y2)
106 grads2 = gradients_impl.gradients(fx2, f_vars + f_side_input, grad_y1)
108 grads3 = gradients_impl.gradients(fx2, f_vars + f_side_input, grad_gy1_y2)
446 grads = gradients_impl.gradients(outputs, inputs + variables, output_grads)
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/ |
| cudnn_rnn_ops_benchmark.py | 32 from tensorflow.python.ops import gradients_impl
113 all_grads = gradients_impl.gradients(
138 gradients = gradients_impl.gradients([outputs, final_state],
162 gradients = gradients_impl.gradients([outputs, final_state],
|
| /external/tensorflow/tensorflow/contrib/nn/python/ops/ |
| fwd_gradients.py | 24 from tensorflow.python.ops.gradients_impl import gradients
|
| /external/tensorflow/tensorflow/python/eager/ |
| tape_test.py | 29 from tensorflow.python.ops import gradients_impl
81 tf_da, tf_db = gradients_impl.gradients(tf_e, [tf_a, tf_b])
143 tf_da, tf_db = gradients_impl.gradients(tf_rr, [tf_a, tf_b])
180 grad = gradients_impl.gradients(f(inp), inp)
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| debug_gradients_test.py | 32 from tensorflow.python.ops import gradients_impl
62 grads = gradients_impl.gradients(y, [self.u, self.v])
93 grads = gradients_impl.gradients(y, [self.u, self.v])
162 gradients_impl.gradients(y, [self.u, self.v])
204 grads = gradients_impl.gradients(y, [self.u, self.v])
230 grads = gradients_impl.gradients(y, [self.u, self.v])
253 grads = gradients_impl.gradients(y, [self.u, self.v])
272 grads = gradients_impl.gradients(y, [self.u, self.v])
|
| /external/tensorflow/tensorflow/contrib/signal/python/kernel_tests/ |
| reconstruction_ops_test.py | 27 from tensorflow.python.ops import gradients_impl
154 gradient = sess.run(gradients_impl.gradients([loss], [signal])[0])
173 gradient = sess.run(gradients_impl.gradients([loss], [signal])[0])
|
