Home
last modified time | relevance | path

Searched refs:gradients_function (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/eager/
Dtape_test.py76 da, db = backprop.gradients_function(fn, [0, 1])(a, b)
96 da, = backprop.gradients_function(forward, ['a'])(aa, bb)
110 da, = backprop.gradients_function(forward, [0])(aa, bb)
139 da, db = backprop.gradients_function(fn, [0, 1])(a, b)
158 grad, = backprop.gradients_function(fn, [0])(logits, labels)
167 g, = backprop.gradients_function(fn, [0])(t)
Dbackprop_test.py75 grad = backprop.gradients_function(fn, [0])(var)[0]
107 grad = backprop.gradients_function(fn, [0])(var)[0]
180 g = backprop.gradients_function(f)
226 grad_fn = backprop.gradients_function(f)
235 self.assertEqual(backprop.gradients_function(f)(int_tensor)[0], None)
249 backprop.gradients_function(f)(constant_op.constant(1.0))
255 (y,) = backprop.gradients_function(lambda x: x * x)(x)
497 g, = backprop.gradients_function(loss, [0])(logits, labels)
533 grad = backprop.gradients_function(first, [0])(x)[0]
537 grad = backprop.gradients_function(second, [0])(f)[0]
[all …]
Dfunction_gradients_test.py363 return backprop.gradients_function(f, [0])(x)[0]
383 return backprop.gradients_function(math_ops.multiply, [0, 1])(x, x)
412 grad_t, = backprop.gradients_function(sq, [0])(t)
419 return backprop.gradients_function(lambda y: y * y, [0])(x)[0]
450 g = backprop.gradients_function(wrapper, [0])(constant_op.constant(0.0))
882 grad_t, = backprop.gradients_function(sq, [0])(t)
898 return backprop.gradients_function(lambda y: y * y, [0])(x)[0]
Dbackprop.py337 def gradients_function(f, params=None): function
Dbenchmarks_test.py388 lambda: backprop.gradients_function(gen_array_ops.identity, [0])(m),
408 self._run(lambda: backprop.gradients_function(lambda x: x, [0])(m), 30000)
/external/tensorflow/tensorflow/python/kernel_tests/
Dreduce_benchmark_test.py59 backprop.gradients_function(math_ops.reduce_sum, [0])(tensor)
68 backprop.gradients_function(math_ops.reduce_sum, [0])(tensor)
Dtensor_array_ops_test.py1015 grad_fn = backprop.gradients_function(func)
1106 grad = backprop.gradients_function(loop)(x)[0]
1578 grad_vals = backprop.gradients_function(func)(
/external/tensorflow/tensorflow/python/ops/
Dgradient_checker_v2.py168 grad_fn_unprep = backprop.gradients_function(f, [param])
/external/tensorflow/tensorflow/compiler/tests/
Deager_test.py163 grad_fn = backprop.gradients_function(f)