Searched defs:jvp_user_function (Results 1 – 2 of 2) sorted by relevance
35 const _jvp_fn_t& jvp_user_function) { in _process_forward_mode_AD()444 const _jvp_fn_t& jvp_user_function, in _wrap_outputs()
633 _jvp_fn_t jvp_user_function = [self]( in _wrap_outputs() local