Searched refs:record_operation (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | custom_gradient.py | 427 tape_lib.record_operation(
482 tape_lib.record_operation(f.__name__, flat_result, recorded_inputs,
|
| D | resource_variable_ops.py | 680 tape.record_operation(
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | tape.py | 181 def record_operation(op_type, output_tensors, input_tensors, backward_function, function
|
| D | forwardprop_test.py | 827 tape_lib.record_operation("ForwardIsSpecial", [d], [c], None,
830 tape_lib.record_operation("ForwardIsSpecial2", [], [], None, lambda: [])
831 tape_lib.record_operation("ForwardIsSpecial3", [e], [d], None,
|
| D | function.py | 846 tape.record_operation(self._inference_function.signature.name,
1028 tape.record_operation(
1049 tape.record_operation(
1316 tape.record_operation(self._forward.signature.name,
|
| D | backprop_test.py | 1035 tape_lib.record_operation('InvalidBackprop', [y], [x], lambda dy: [])
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | func_graph.py | 666 tape.record_operation("captured_value", [placeholder], [tensor],
717 tape.record_operation("captured_value", [placeholder], [variable],
738 tape.record_operation("captured_value", [graph_const], [tensor],
|
| D | ops.py | 1175 tape.record_operation("_copy", [new_tensor], [self], grad_fun)
1489 tape.record_operation("pack_eager_tensors", [packed_tensor], tensors,
|