Searched refs:batch_map_config (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/tests/ut/python/dataset/ |
| D | test_var_batch_map.py | 162 def batch_map_config(num, r, batch_size, func, res): function
169 batch_map_config(4, 2, 2, invert_sign_per_epoch, tst1)
173 batch_map_config(4, 2, 2, invert_sign_per_batch, tst2)
201 def batch_map_config(num, r, batch_size, func, col_names, res): function
208 batch_map_config(4, 2, 2, invert_sign_per_batch, ["num_square"], tst1)
212 batch_map_config(4, 2, 2, invert_sign_per_batch_multi_col, ["num", "num_square"], tst2)
218 batch_map_config(4, 3, 2, col1_col2_add_num, ["num_square", "num"], tst3)
222 batch_map_config(4, 3, 2, col1_col2_add_num, ["num", "num_square"], tst4)
253 def batch_map_config(num, r, fbatch, fmap, col_names, res): function
263 batch_map_config(10, 1, batch_func, map_func, ["col1", "col2"], tst1)
[all …]
|