Searched refs:tape_lib (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/framework/experimental/ |
| D | unified_api_test.py | 31 from tensorflow.python.framework.experimental import tape as tape_lib unknown
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:
447 tape_lib,
477 tape_lib,
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | custom_gradient.py | 19 from tensorflow.python.eager import tape as tape_lib unknown
420 with tape_lib.VariableWatcher() as variable_watcher:
520 tape_lib.record_operation(
530 with tape_lib.VariableWatcher() as variable_watcher:
575 tape_lib.record_operation(f.__name__, flat_result, recorded_inputs,
691 with tape_lib.stop_recording():
|
| D | script_ops.py | 29 from tensorflow.python.eager import tape as tape_lib unknown
136 self._support_graph_mode_gradient or tape_lib.could_possibly_record())
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | forwardprop_test.py | 30 from tensorflow.python.eager import tape as tape_lib unknown
739 self.assertFalse(tape_lib.should_record_backprop([c]))
743 self.assertTrue(tape_lib.should_record_backprop([c]))
744 with tape_lib.stop_recording():
747 self.assertFalse(tape_lib.should_record_backprop([c]))
750 self.assertTrue(tape_lib.should_record_backprop([c]))
751 self.assertFalse(tape_lib.should_record_backprop([d]))
762 with tape_lib.stop_recording():
771 tape_lib.record_operation_forwardprop_only(
774 tape_lib.record_operation_backprop_only("CustomBackwardMul", [e],
[all …]
|
| D | backprop_test.py | 25 from tensorflow.python.eager import tape as tape_lib unknown
1084 tape_lib.record_operation('InvalidBackprop', [y], [x], lambda dy: [])
|