Searched refs:ForwardAccumulator (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/python/eager/ |
| D | forwardprop_test.py | 61 with forwardprop.ForwardAccumulator(primals, tangents) as acc:
131 with forwardprop.ForwardAccumulator(
146 with forwardprop.ForwardAccumulator(primals, tangents) as acc:
301 with forwardprop.ForwardAccumulator(x, 2.) as acc1:
302 with forwardprop.ForwardAccumulator(x, 2.) as acc2:
312 with forwardprop.ForwardAccumulator(x, 2.) as acc:
322 with forwardprop.ForwardAccumulator(x, 2.) as acc:
342 with forwardprop.ForwardAccumulator(x, array_ops.ones_like(x)) as acc:
350 with forwardprop.ForwardAccumulator(x, xt) as acc:
359 with forwardprop.ForwardAccumulator([x, x], [1., 2.]):
[all …]
|
| D | forwardprop.py | 222 class ForwardAccumulator(): class
465 acc = super(ForwardAccumulator, cls).__new__(cls, primals, tangents)
|
| D | benchmarks_test.py | 826 with forwardprop.ForwardAccumulator(m, tangent) as acc:
839 with forwardprop.ForwardAccumulator(x, tangent) as acc:
858 with forwardprop.ForwardAccumulator(x, tangent) as acc:
878 with forwardprop.ForwardAccumulator(m, tangent) as acc:
|
| D | pywrap_tfe_src.cc | 1561 typedef tensorflow::eager::ForwardAccumulator<PyObject, PyBackwardFunction,
1563 ForwardAccumulator; typedef
1614 ForwardAccumulator* accumulator;
2842 accumulator->accumulator = new ForwardAccumulator(*py_vspace, use_batch);
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.autodiff.-forward-accumulator.pbtxt | 1 path: "tensorflow.autodiff.ForwardAccumulator"
3 is_instance: "<class \'tensorflow.python.eager.forwardprop.ForwardAccumulator\'>"
|
| D | tensorflow.autodiff.pbtxt | 4 name: "ForwardAccumulator"
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | tape.h | 228 class ForwardAccumulator {
232 explicit ForwardAccumulator( in ForwardAccumulator() function
239 virtual ~ForwardAccumulator() { in ~ForwardAccumulator()
900 bool ForwardAccumulator<Gradient, BackwardFunction, TapeTensor>::ShouldRecord( in ShouldRecord()
924 ForwardAccumulator<Gradient, BackwardFunction, TapeTensor>::ForwardpropFromTape( in ForwardpropFromTape()
1022 Status ForwardAccumulator<Gradient, BackwardFunction, TapeTensor>::Accumulate( in Accumulate()
1116 void ForwardAccumulator<Gradient, BackwardFunction, TapeTensor>::Watch( in Watch()
1134 void ForwardAccumulator<Gradient, BackwardFunction, TapeTensor>::DeleteGradient( in DeleteGradient()
1144 Gradient* ForwardAccumulator<Gradient, BackwardFunction, TapeTensor>::FetchJVP( in FetchJVP()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients.py | 24 from tensorflow.python.eager.forwardprop import ForwardAccumulator
|
| /external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/ |
| D | hvp_test.py | 34 with forwardprop.ForwardAccumulator(
47 with forwardprop.ForwardAccumulator(
|
| /external/tensorflow/tensorflow/python/keras/integration_test/ |
| D | forwardprop_test.py | 29 with tf.autodiff.ForwardAccumulator(primals, tangents) as acc:
74 with tf.autodiff.ForwardAccumulator(primals, tangents) as acc:
285 with tf.autodiff.ForwardAccumulator(parameters, tangents):
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_interop_test.py | 379 with tf.autodiff.ForwardAccumulator(x, xt) as acc:
|
| /external/tensorflow/ |
| D | RELEASE.md | 1615 * Added `tf.autodiff.ForwardAccumulator` for forward-mode autodiff
|