Searched refs:_fetch_jvp (Results 1 – 1 of 1) sorted by relevance
435 def _fetch_jvp(tensor): function446 return nest.map_structure(_fetch_jvp, primals)