Searched refs:record_operation (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | custom_gradient.py | 248 tape_lib.record_operation(
294 tape_lib.record_operation(f.__name__, flat_result, input_tensors,
|
| D | resource_variable_ops.py | 864 tape.record_operation(
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | tape.py | 132 def record_operation(op_type, output_tensors, input_tensors, backward_function): function
|
| D | function.py | 852 tape.record_operation(self._forward_function.signature.name, real_outputs,
890 tape.record_operation(self._inference_function.signature.name, outputs,
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | func_graph.py | 500 tape.record_operation("captured_value", [captured_tensor], [tensor],
|
| D | ops.py | 860 tape.record_operation("_copy", [new_tensor], [self], grad_fun)
|