Home
last modified time | relevance | path

Searched defs:param_jit_level (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/tests/st/ops/
Dtest_any.py185 def test_any_vmap(param_jit_level): argument
Dtest_ops_masked_fill.py262 def test_ops_masked_fill_forward_with_broadcast(context_mode, param_jit_level): argument
Dtest_ops_gather_ext.py153 def test_gather_ext_vmap(param_jit_level): argument
Dtest_ops_where.py263 def test_where_vmap(param_jit_level): argument
Dtest_ops_one_hot_ext.py275 def test_onehot_vmap(param_jit_level): argument