Home
last modified time | relevance | path

Searched refs:record_operation (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dcustom_gradient.py427 tape_lib.record_operation(
482 tape_lib.record_operation(f.__name__, flat_result, recorded_inputs,
Dresource_variable_ops.py680 tape.record_operation(
/external/tensorflow/tensorflow/python/eager/
Dtape.py181 def record_operation(op_type, output_tensors, input_tensors, backward_function, function
Dforwardprop_test.py827 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,
Dfunction.py846 tape.record_operation(self._inference_function.signature.name,
1028 tape.record_operation(
1049 tape.record_operation(
1316 tape.record_operation(self._forward.signature.name,
Dbackprop_test.py1035 tape_lib.record_operation('InvalidBackprop', [y], [x], lambda dy: [])
/external/tensorflow/tensorflow/python/framework/
Dfunc_graph.py666 tape.record_operation("captured_value", [placeholder], [tensor],
717 tape.record_operation("captured_value", [placeholder], [variable],
738 tape.record_operation("captured_value", [graph_const], [tensor],
Dops.py1175 tape.record_operation("_copy", [new_tensor], [self], grad_fun)
1489 tape.record_operation("pack_eager_tensors", [packed_tensor], tensors,