Searched defs:param_jit_level (Results 1 – 5 of 5) sorted by relevance
185 def test_any_vmap(param_jit_level): argument
262 def test_ops_masked_fill_forward_with_broadcast(context_mode, param_jit_level): argument
153 def test_gather_ext_vmap(param_jit_level): argument
263 def test_where_vmap(param_jit_level): argument
275 def test_onehot_vmap(param_jit_level): argument