| /external/tensorflow/tensorflow/python/eager/ |
| D | backprop_test.py | 144 with backprop.GradientTape() as t:
166 with backprop.GradientTape() as t:
186 with backprop.GradientTape() as t:
207 with backprop.GradientTape() as t:
213 with backprop.GradientTape() as t:
317 with backprop.GradientTape() as t:
324 with backprop.GradientTape() as t:
335 with backprop.GradientTape() as t:
344 with backprop.GradientTape() as t:
351 with backprop.GradientTape() as t:
[all …]
|
| D | function_gradients_test.py | 108 with backprop.GradientTape(persistent=persistent) as tape:
134 with backprop.GradientTape() as tape:
158 with backprop.GradientTape() as tape:
186 with backprop.GradientTape(persistent=True) as tape:
203 with backprop.GradientTape(persistent=True) as tape:
217 with backprop.GradientTape() as t:
220 with backprop.GradientTape() as tt:
280 with backprop.GradientTape() as t:
295 with backprop.GradientTape() as t:
458 with backprop.GradientTape() as tp:
[all …]
|
| D | forwardprop_test.py | 114 with backprop.GradientTape() as tape:
147 with backprop.GradientTape() as tape:
674 with backprop.GradientTape(persistent=True) as g:
676 with backprop.GradientTape() as gg:
694 with backprop.GradientTape() as tape:
716 with backprop.GradientTape(persistent=True) as tape:
765 gradient_tape = backprop.GradientTape()
767 gradient_tape = backprop.GradientTape()
796 gradient_tape = backprop.GradientTape()
798 gradient_tape = backprop.GradientTape()
[all …]
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | gradients.h | 129 class Tape : protected eager::GradientTape<AbstractTensorHandle,
132 using GradientTape<AbstractTensorHandle, GradientFunction,
133 TapeTensor>::GradientTape;
136 using GradientTape<AbstractTensorHandle, GradientFunction,
|
| D | tape.h | 126 class GradientTape {
132 explicit GradientTape(bool persistent) : persistent_(persistent) {} in GradientTape() function
133 ~GradientTape() { in ~GradientTape()
347 GradientTape<Gradient, BackwardFunction, TapeTensor>* backward_tape, in AccumulatorCallState()
356 GradientTape<Gradient, BackwardFunction, TapeTensor>* backward_tape;
385 bool GradientTape<Gradient, BackwardFunction, TapeTensor>::ShouldRecord( in ShouldRecord()
400 void GradientTape<Gradient, BackwardFunction, TapeTensor>::Watch( in Watch()
406 void GradientTape<Gradient, BackwardFunction, TapeTensor>::RecordOperation( in RecordOperation()
437 void GradientTape<Gradient, BackwardFunction, TapeTensor>::DeleteTrace( in DeleteTrace()
682 Status GradientTape<Gradient, BackwardFunction, TapeTensor>::ComputeGradient( in ComputeGradient()
[all …]
|
| /external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/ |
| D | hvp_test.py | 36 with tf.GradientTape() as grad_tape:
45 with tf.GradientTape() as grad_tape:
56 with tf.GradientTape() as grad_tape:
68 with tf.GradientTape() as outer_tape:
69 with tf.GradientTape() as inner_tape:
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | custom_training_loop_models_test.py | 79 with backprop.GradientTape() as tape:
104 with backprop.GradientTape() as tape:
142 with backprop.GradientTape() as tape:
167 with backprop.GradientTape() as tape:
199 with backprop.GradientTape() as tape:
243 with backprop.GradientTape() as tape:
297 with backprop.GradientTape() as tape:
314 with backprop.GradientTape() as tape:
352 with backprop.GradientTape() as tape:
379 with backprop.GradientTape() as tape:
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/python/ |
| D | test_utils.py | 32 with backprop.GradientTape() as gt:
39 with backprop.GradientTape() as gt:
|
| /external/tensorflow/tensorflow/python/framework/experimental/ |
| D | unified_api_test.py | 91 with tape_lib.GradientTape() as tape:
145 with tape_lib.GradientTape() as tape:
195 with tape_lib.GradientTape() as tape:
240 with tape_lib.GradientTape() as tape:
289 with tape_lib.GradientTape() as tape:
337 with tape_lib.GradientTape() as tape:
382 with tape_lib.GradientTape() as tape:
411 with backprop_lib.GradientTape() as tape:
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | map_ops_test.py | 190 with backprop.GradientTape() as tape:
202 with backprop.GradientTape(persistent=True) as tape:
228 with backprop.GradientTape() as tape:
243 with backprop.GradientTape(persistent=True) as tape:
266 with backprop.GradientTape(persistent=True) as tape:
287 with backprop.GradientTape(persistent=True) as tape:
308 with backprop.GradientTape(persistent=True) as tape:
327 with backprop.GradientTape(persistent=True) as tape:
348 with backprop.GradientTape(persistent=True) as tape:
366 with backprop.GradientTape(persistent=True) as tape:
[all …]
|
| D | relu_op_test.py | 151 with backprop.GradientTape() as tape:
169 with backprop.GradientTape() as tape:
197 with backprop.GradientTape() as tape:
340 with backprop.GradientTape() as tape:
358 with backprop.GradientTape() as tape:
453 with backprop.GradientTape(persistent=True) as tape:
470 with backprop.GradientTape() as tape:
488 with backprop.GradientTape() as tape:
554 with backprop.GradientTape() as tape:
572 with backprop.GradientTape() as tape:
|
| /external/tensorflow/tensorflow/python/keras/tests/ |
| D | memory_test.py | 56 with backprop.GradientTape():
69 with backprop.GradientTape() as tape:
|
| D | tracking_util_test.py | 116 with backprop.GradientTape() as tape:
124 with backprop.GradientTape() as tape:
206 with backprop.GradientTape() as tape:
283 with backprop.GradientTape() as tape:
305 with backprop.GradientTape() as tape:
342 with backprop.GradientTape() as tape:
414 with backprop.GradientTape() as tape:
458 with backprop.GradientTape() as tape:
622 with backprop.GradientTape() as tape:
653 with backprop.GradientTape() as tape:
[all …]
|
| D | custom_training_loop_test.py | 77 with backprop.GradientTape() as tape:
101 with backprop.GradientTape() as tape:
124 with backprop.GradientTape() as tape:
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | custom_training_loop_gradient_test.py | 80 with backprop.GradientTape() as tape:
107 with backprop.GradientTape() as tape:
140 with backprop.GradientTape() as tape:
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.-gradient-tape.pbtxt | 1 path: "tensorflow.GradientTape"
3 is_instance: "<class \'tensorflow.python.eager.backprop.GradientTape\'>"
|
| D | tensorflow.autodiff.-gradient-tape.pbtxt | 1 path: "tensorflow.autodiff.GradientTape"
3 is_instance: "<class \'tensorflow.python.eager.backprop.GradientTape\'>"
|
| D | tensorflow.autodiff.pbtxt | 8 name: "GradientTape"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-gradient-tape.pbtxt | 1 path: "tensorflow.GradientTape"
3 is_instance: "<class \'tensorflow.python.eager.backprop.GradientTape\'>"
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops_test.py | 40 with backprop.GradientTape() as tape:
44 with backprop.GradientTape() as tape:
|
| /external/tensorflow/tensorflow/python/keras/integration_test/ |
| D | forwardprop_test.py | 61 with tf.GradientTape() as tape:
75 with tf.GradientTape() as tape:
211 with tf.GradientTape() as tape:
252 with tf.GradientTape() as tape:
258 with tf.GradientTape(persistent=True) as tape:
|
| D | gradients_test.py | 49 with tf.GradientTape(persistent=True) as tape:
96 with tf.GradientTape() as tape:
|
| /external/tensorflow/tensorflow/python/eager/memory_tests/ |
| D | memory_test.py | 80 with backprop.GradientTape() as t1:
82 with backprop.GradientTape() as t2:
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale_benchmark.py | 107 with backprop.GradientTape() as tape:
113 with backprop.GradientTape() as tape:
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | eager_test.py | 61 with backprop.GradientTape(persistent=True) as tape:
267 with backprop.GradientTape() as tape:
281 with backprop.GradientTape() as tape:
491 with backprop.GradientTape() as tape:
505 with backprop.GradientTape() as tape:
522 with backprop.GradientTape() as tape:
580 with backprop.GradientTape() as tape:
603 with backprop.GradientTape(persistent=True) as tape:
|