Searched refs:tape_lib (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/framework/experimental/ |
| D | unified_api_test.py | 35 from tensorflow.python.framework.experimental import tape as tape_lib unknown
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:
450 tape_lib,
480 tape_lib,
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | custom_gradient.py | 22 from tensorflow.python.eager import tape as tape_lib unknown
329 with tape_lib.VariableWatcher() as variable_watcher:
427 tape_lib.record_operation(
437 with tape_lib.VariableWatcher() as variable_watcher:
482 tape_lib.record_operation(f.__name__, flat_result, recorded_inputs,
515 with tape_lib.stop_recording():
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | forwardprop_test.py | 33 from tensorflow.python.eager import tape as tape_lib unknown
695 self.assertFalse(tape_lib.should_record_backprop([c]))
699 self.assertTrue(tape_lib.should_record_backprop([c]))
700 with tape_lib.stop_recording():
703 self.assertFalse(tape_lib.should_record_backprop([c]))
706 self.assertTrue(tape_lib.should_record_backprop([c]))
707 self.assertFalse(tape_lib.should_record_backprop([d]))
718 with tape_lib.stop_recording():
727 tape_lib.record_operation_forwardprop_only(
730 tape_lib.record_operation_backprop_only("CustomBackwardMul", [e],
[all …]
|
| D | backprop_test.py | 29 from tensorflow.python.eager import tape as tape_lib unknown
1035 tape_lib.record_operation('InvalidBackprop', [y], [x], lambda dy: [])
|