Searched refs:AutogradCompilerCall (Results 1 – 2 of 2) sorted by relevance
189 struct AutogradCompilerCall { struct531 CompiledNodeArgs(AutogradCompilerCall& compiler, NodeCall& node_call) in CompiledNodeArgs()557 AutogradCompilerCall& _compiler;772 AutogradCompilerCall& c, in SwapSavedVariables()827 AutogradCompilerCall& compiler;
212 AutogradCompilerCall& call, in check_dynamic_sizes()412 AutogradCompilerCall& compiler_call, in call_begin_capture()478 AutogradCompilerCall compiler_call; in _compiled_autograd_impl()