Searched refs:GradientTape (Results 1 – 25 of 84) sorted by relevance
1234
| /external/tensorflow/tensorflow/python/eager/ |
| D | backprop_test.py | 105 with backprop.GradientTape() as t:
125 with backprop.GradientTape() as t:
144 with backprop.GradientTape() as t:
150 with backprop.GradientTape() as t:
252 with backprop.GradientTape() as t:
259 with backprop.GradientTape() as t:
266 with backprop.GradientTape() as t:
273 with backprop.GradientTape() as t:
281 with backprop.GradientTape() as t:
291 with backprop.GradientTape() as t:
[all …]
|
| D | function_gradients_test.py | 121 with backprop.GradientTape() as t:
136 with backprop.GradientTape() as t:
282 with backprop.GradientTape() as tp:
302 with backprop.GradientTape() as tp:
328 with backprop.GradientTape() as tp:
352 with backprop.GradientTape() as tp:
361 with backprop.GradientTape() as tp:
369 with backprop.GradientTape() as tp:
376 with backprop.GradientTape() as tp:
399 with backprop.GradientTape() as tp:
[all …]
|
| D | pywrap_tfe_test.py | 79 with backprop.GradientTape(persistent=True) as tape:
93 with backprop.GradientTape(persistent=True) as tape:
126 with backprop.GradientTape(persistent=True) as tape:
159 with backprop.GradientTape(persistent=True) as tape:
|
| D | memory_test.py | 103 with backprop.GradientTape():
116 with backprop.GradientTape() as tape:
|
| D | benchmarks_test.py | 306 with backprop.GradientTape() as tape:
314 with backprop.GradientTape():
395 with backprop.GradientTape() as gt:
410 with backprop.GradientTape() as tape:
416 with backprop.GradientTape() as tape:
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | relu_op_test.py | 126 with backprop.GradientTape() as tape:
175 with backprop.GradientTape() as tape:
194 with backprop.GradientTape() as tape:
348 with backprop.GradientTape() as tape:
368 with backprop.GradientTape() as tape:
448 with backprop.GradientTape(persistent=True) as tape:
465 with backprop.GradientTape() as tape:
484 with backprop.GradientTape() as tape:
553 with backprop.GradientTape() as tape:
572 with backprop.GradientTape() as tape:
|
| D | list_ops_test.py | 126 with backprop.GradientTape() as tape:
299 with backprop.GradientTape() as tape:
401 with backprop.GradientTape(persistent=True) as tape:
522 with backprop.GradientTape() as tape:
535 with backprop.GradientTape() as tape:
593 with backprop.GradientTape() as tape:
861 with backprop.GradientTape() as tape:
872 with backprop.GradientTape() as tape:
883 with backprop.GradientTape() as tape:
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | tape.h | 111 class GradientTape {
117 GradientTape(bool persistent) : persistent_(persistent) {} in GradientTape() function
118 ~GradientTape() { in ~GradientTape()
185 bool GradientTape<Gradient, BackwardFunction, TapeTensor>::ShouldRecord( in ShouldRecord()
200 void GradientTape<Gradient, BackwardFunction, TapeTensor>::Watch( in Watch()
206 void GradientTape<Gradient, BackwardFunction, TapeTensor>::RecordOperation( in RecordOperation()
237 void GradientTape<Gradient, BackwardFunction, TapeTensor>::DeleteTrace( in DeleteTrace()
475 Status GradientTape<Gradient, BackwardFunction, TapeTensor>::ComputeGradient( in ComputeGradient()
|
| /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.-gradient-tape.pbtxt | 1 path: "tensorflow.GradientTape"
3 is_instance: "<class \'tensorflow.python.eager.backprop.GradientTape\'>"
|
| /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:
440 with backprop.GradientTape() as tape:
454 with backprop.GradientTape() as tape:
471 with backprop.GradientTape() as tape:
529 with backprop.GradientTape() as tape:
552 with backprop.GradientTape(persistent=True) as tape:
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/ |
| D | blocks.py | 189 with tf.GradientTape() as gtape:
201 with tf.GradientTape() as ftape:
225 with tf.GradientTape() as gtape:
234 with tf.GradientTape() as x1tape:
239 with tf.GradientTape() as ftape:
247 with tf.GradientTape() as x2tape:
|
| D | ops_test.py | 46 with tfe.GradientTape() as tape:
69 with tfe.GradientTape() as tape:
|
| D | blocks_test.py | 144 with tf.GradientTape() as tape:
173 with tf.GradientTape() as tape:
207 with tf.GradientTape() as tape:
247 with tf.GradientTape() as tape:
|
| D | revnet.py | 155 with tf.GradientTape() as tape:
175 with tf.GradientTape() as tape:
|
| /external/tensorflow/tensorflow/contrib/eager/python/ |
| D | tfe.py | 151 GradientTape = backprop.GradientTape # pylint: disable=invalid-name variable
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients.py | 23 from tensorflow.python.eager.backprop import GradientTape
|
| D | custom_gradient.py | 182 with backprop.GradientTape() as tape:
258 with backprop.GradientTape() as tape:
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/densenet/ |
| D | densenet_test.py | 126 with tf.GradientTape() as tape_true, tf.GradientTape() as tape_keras:
141 with tf.GradientTape() as tape:
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | util_test.py | 270 with backprop.GradientTape() as tape:
278 with backprop.GradientTape() as tape:
404 with backprop.GradientTape() as tape:
481 with backprop.GradientTape() as tape:
503 with backprop.GradientTape() as tape:
538 with backprop.GradientTape() as tape:
642 with backprop.GradientTape() as tape:
728 with backprop.GradientTape() as tape:
1184 with backprop.GradientTape() as tape:
1215 with backprop.GradientTape() as tape:
[all …]
|
| D | util_xla_test.py | 72 with backprop.GradientTape() as tape:
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_eager.py | 26 from tensorflow.python.eager.backprop import GradientTape
208 with GradientTape() as tape:
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/gan/ |
| D | mnist.py | 232 with tf.GradientTape() as gen_tape, tf.GradientTape() as disc_tape:
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | embeddings_test.py | 91 with backprop.GradientTape() as tape:
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | array_test.py | 226 with backprop.GradientTape(persistent=True) as g:
259 with backprop.GradientTape(persistent=True) as g:
|
1234