| /external/tensorflow/tensorflow/python/eager/ |
| D | backprop_test.py | 140 with backprop.GradientTape() as t:
163 with backprop.GradientTape() as tape:
177 with backprop.GradientTape() as t:
197 with backprop.GradientTape() as t:
218 with backprop.GradientTape() as t:
241 with backprop.GradientTape() as tape:
248 with backprop.GradientTape() as t:
352 with backprop.GradientTape() as t:
359 with backprop.GradientTape() as t:
370 with backprop.GradientTape() as t:
[all …]
|
| D | function_gradients_test.py | 105 with backprop.GradientTape(persistent=persistent) as tape:
131 with backprop.GradientTape() as tape:
155 with backprop.GradientTape() as tape:
183 with backprop.GradientTape(persistent=True) as tape:
200 with backprop.GradientTape(persistent=True) as tape:
214 with backprop.GradientTape() as t:
217 with backprop.GradientTape() as tt:
277 with backprop.GradientTape() as t:
292 with backprop.GradientTape() as t:
455 with backprop.GradientTape() as tp:
[all …]
|
| D | forwardprop_test.py | 115 with backprop.GradientTape() as tape:
148 with backprop.GradientTape() as tape:
718 with backprop.GradientTape(persistent=True) as g:
720 with backprop.GradientTape() as gg:
738 with backprop.GradientTape() as tape:
760 with backprop.GradientTape(persistent=True) as tape:
809 gradient_tape = backprop.GradientTape()
811 gradient_tape = backprop.GradientTape()
840 gradient_tape = backprop.GradientTape()
842 gradient_tape = backprop.GradientTape()
[all …]
|
| D | pywrap_tfe_test.py | 90 with backprop.GradientTape(persistent=True) as tape:
106 with backprop.GradientTape(persistent=True) as tape:
142 with backprop.GradientTape(persistent=True) as tape:
177 with backprop.GradientTape(persistent=True) as tape:
|
| /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 …]
|
| D | gradients.cc | 223 GradientTape::Watch(ToId(t)); in Watch()
240 GradientTape::RecordOperation( in RecordOperation()
257 return GradientTape::ShouldRecord(tensor_ids, tensor_dtypes); in ShouldRecord()
260 GradientTape::DeleteTrace(ToId(t)); in DeleteTrace()
292 TF_RETURN_IF_ERROR(GradientTape::ComputeGradient( in ComputeGradient()
|
| /external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/ |
| D | hvp_test.py | 31 with tf.GradientTape() as grad_tape:
40 with tf.GradientTape() as grad_tape:
51 with tf.GradientTape() as grad_tape:
63 with tf.GradientTape() as outer_tape:
64 with tf.GradientTape() as inner_tape:
|
| /external/tensorflow/tensorflow/python/kernel_tests/data_structures/ |
| D | map_ops_test.py | 186 with backprop.GradientTape() as tape:
198 with backprop.GradientTape(persistent=True) as tape:
224 with backprop.GradientTape() as tape:
239 with backprop.GradientTape(persistent=True) as tape:
262 with backprop.GradientTape(persistent=True) as tape:
283 with backprop.GradientTape(persistent=True) as tape:
304 with backprop.GradientTape(persistent=True) as tape:
323 with backprop.GradientTape(persistent=True) as tape:
344 with backprop.GradientTape(persistent=True) as tape:
362 with backprop.GradientTape(persistent=True) as tape:
[all …]
|
| /external/tensorflow/tensorflow/python/framework/experimental/ |
| D | unified_api_test.py | 88 with tape_lib.GradientTape() as tape:
142 with tape_lib.GradientTape() as tape:
192 with tape_lib.GradientTape() as tape:
237 with tape_lib.GradientTape() as tape:
286 with tape_lib.GradientTape() as tape:
334 with tape_lib.GradientTape() as tape:
379 with tape_lib.GradientTape() as tape:
408 with backprop_lib.GradientTape() as tape:
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/python/ |
| D | test_utils.py | 28 with backprop.GradientTape() as gt:
35 with backprop.GradientTape() as gt:
|
| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | relu_op_test.py | 147 with backprop.GradientTape() as tape:
165 with backprop.GradientTape() as tape:
193 with backprop.GradientTape() as tape:
339 with backprop.GradientTape() as tape:
357 with backprop.GradientTape() as tape:
452 with backprop.GradientTape(persistent=True) as tape:
469 with backprop.GradientTape() as tape:
487 with backprop.GradientTape() as tape:
553 with backprop.GradientTape() as tape:
571 with backprop.GradientTape() as tape:
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | custom_training_loop_gradient_test.py | 76 with backprop.GradientTape() as tape:
103 with backprop.GradientTape() as tape:
136 with backprop.GradientTape() as tape:
|
| /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/tools/api/golden/v2/ |
| D | tensorflow.autodiff.-gradient-tape.pbtxt | 1 path: "tensorflow.autodiff.GradientTape"
3 is_instance: "<class \'tensorflow.python.eager.backprop.GradientTape\'>"
|
| D | tensorflow.-gradient-tape.pbtxt | 1 path: "tensorflow.GradientTape"
3 is_instance: "<class \'tensorflow.python.eager.backprop.GradientTape\'>"
|
| D | tensorflow.autodiff.pbtxt | 8 name: "GradientTape"
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_grad_d9m_test.py | 78 with backprop.GradientTape() as tape:
149 with backprop.GradientTape(persistent=True) as tape:
222 with backprop.GradientTape(persistent=True) as tape:
322 with backprop.GradientTape(persistent=True) as tape:
|
| D | array_ops_test.py | 38 with backprop.GradientTape() as tape:
42 with backprop.GradientTape() as tape:
|
| D | gradients.py | 19 from tensorflow.python.eager.backprop import GradientTape
|
| /external/tensorflow/tensorflow/python/eager/memory_tests/ |
| D | memory_test.py | 76 with backprop.GradientTape() as t1:
78 with backprop.GradientTape() as t2:
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | eager_test.py | 58 with backprop.GradientTape(persistent=True) as tape:
264 with backprop.GradientTape() as tape:
278 with backprop.GradientTape() as tape:
488 with backprop.GradientTape() as tape:
502 with backprop.GradientTape() as tape:
519 with backprop.GradientTape() as tape:
577 with backprop.GradientTape() as tape:
600 with backprop.GradientTape(persistent=True) as tape:
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | linear_operator_toeplitz_test.py | 151 with backprop.GradientTape() as tape:
154 with backprop.GradientTape() as tape:
|
| /external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/ |
| D | sparse_xent_op_test_base.py | 39 with backprop_lib.GradientTape() as tape:
244 with backprop_lib.GradientTape() as tape:
274 with backprop_lib.GradientTape() as tape:
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_interop_test.py | 84 with tf.GradientTape() as t:
103 with tf.GradientTape() as t:
342 with tf.GradientTape() as g:
357 with tf.GradientTape() as g:
|