Searched refs:compute_quantities_for_vmap_test (Results 1 – 2 of 2) sorted by relevance
348 def compute_quantities_for_vmap_test( function
23 compute_quantities_for_vmap_test,4061 for vmap_out, loop_out in compute_quantities_for_vmap_test(4082 for _ in compute_quantities_for_vmap_test(4093 for vmap_out, loop_out in compute_quantities_for_vmap_test(