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.py73 da, db = backprop.gradients_function(fn, [0, 1])(a, b)
93 da, = backprop.gradients_function(forward, ['a'])(aa, bb)
107 da, = backprop.gradients_function(forward, [0])(aa, bb)
136 da, db = backprop.gradients_function(fn, [0, 1])(a, b)
155 grad, = backprop.gradients_function(fn, [0])(logits, labels)
164 g, = backprop.gradients_function(fn, [0])(t)
Dbackprop_test.py71 grad = backprop.gradients_function(fn, [0])(var)[0]
103 grad = backprop.gradients_function(fn, [0])(var)[0]
191 g = backprop.gradients_function(f)
261 grad_fn = backprop.gradients_function(f)
270 self.assertEqual(backprop.gradients_function(f)(int_tensor)[0], None)
284 backprop.gradients_function(f)(constant_op.constant(1.0))
290 (y,) = backprop.gradients_function(lambda x: x * x)(x)
532 g, = backprop.gradients_function(loss, [0])(logits, labels)
568 grad = backprop.gradients_function(first, [0])(x)[0]
572 grad = backprop.gradients_function(second, [0])(f)[0]
[all …]
Dfunction_gradients_test.py360 return backprop.gradients_function(f, [0])(x)[0]
380 return backprop.gradients_function(math_ops.multiply, [0, 1])(x, x)
409 grad_t, = backprop.gradients_function(sq, [0])(t)
416 return backprop.gradients_function(lambda y: y * y, [0])(x)[0]
447 g = backprop.gradients_function(wrapper, [0])(constant_op.constant(0.0))
879 grad_t, = backprop.gradients_function(sq, [0])(t)
895 return backprop.gradients_function(lambda y: y * y, [0])(x)[0]
Dbackprop.py345 def gradients_function(f, params=None): function
Dbenchmarks_test.py380 lambda: backprop.gradients_function(gen_array_ops.identity, [0])(m),
400 self._run(lambda: backprop.gradients_function(lambda x: x, [0])(m), 30000)
/external/tensorflow/tensorflow/python/kernel_tests/math_ops/
Dreduce_benchmark_test.py54 backprop.gradients_function(math_ops.reduce_sum, [0])(tensor)
63 backprop.gradients_function(math_ops.reduce_sum, [0])(tensor)
/external/tensorflow/tensorflow/python/ops/
Dgradient_checker_v2.py166 grad_fn_unprep = backprop.gradients_function(f, [param])
/external/tensorflow/tensorflow/python/kernel_tests/data_structures/
Dtensor_array_ops_test.py1011 grad_fn = backprop.gradients_function(func)
1102 grad = backprop.gradients_function(loop)(x)[0]
1574 grad_vals = backprop.gradients_function(func)(
/external/tensorflow/tensorflow/compiler/tests/
Deager_test.py160 grad_fn = backprop.gradients_function(f)