Searched defs:saved_tensors (Results 1 – 7 of 7) sorted by relevance
97 def call_backward(backward_c_function, saved_tensors, *args): argument
154 saved_tensors, argument
438 def jvp_no_context(saved_tensors, tangents): argument592 def saved_tensors(self): member in CtxWithSavedTensors
815 def saved_tensors(self): member in NestedIOFunction
794 def get_args(saved_tensors): argument
234 THPObjectPtr saved_tensors(unpack_saved_variables( in defer_to_dynamo() local
830 saved_tensors=None, argument