Searched refs:record_operation_forwardprop_only (Results 1 – 3 of 3) sorted by relevance
197 def record_operation_forwardprop_only(op_type, output_tensors, input_tensors, function
727 tape_lib.record_operation_forwardprop_only(815 tape_lib.record_operation_forwardprop_only("FunctionWithInlineJVPs",
1078 tape.record_operation_forwardprop_only(1310 tape.record_operation_forwardprop_only(