Searched refs:primals_and_nztangents (Results 1 – 1 of 1) sorted by relevance
645 def f_jvp_traceable(nonzeros, *primals_and_nztangents): argument647 primals = list(primals_and_nztangents[:num_primals])648 nonzero_tangents = iter(primals_and_nztangents[num_primals:])