Searched defs:_foreach_run (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ |
| D | test_ops_gather_ext.py | 159 def _foreach_run(x, dim, index, batch_axis): function
200 def _foreach_run(inputs, dim, indices): function
|
| D | test_any.py | 191 def _foreach_run(inputs, batch): function
|
| D | test_ops_where.py | 269 def _foreach_run(condition, x, y, batch): function
|
| D | test_ops_one_hot_ext.py | 281 def _foreach_run(inputs, num_classes, batch): function
|