Searched refs:per_batch_map (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/tests/ut/python/dataset/ |
| D | test_var_batch_map.py | 91 … .batch(batch_size=func, drop_remainder=drop, input_columns=["num"], per_batch_map=simple_copy)
106 ….batch(batch_size=func, drop_remainder=drop, input_columns=["num"], per_batch_map=simple_copy).rep…
164 .batch(batch_size=batch_size, input_columns=["num"], per_batch_map=func).repeat(r)
203 .batch(batch_size=batch_size, input_columns=col_names, per_batch_map=func).repeat(r)
255 .batch(batch_size=fbatch, input_columns=col_names, per_batch_map=fmap).repeat(r)
277 …tch(batch_size=add_one, drop_remainder=True, input_columns=["image"], per_batch_map=np_psedo_resiz…
306 …torDataset((lambda: gen(100)), ["num"]).batch(4, input_columns=["num"], per_batch_map=bad_map_func)
335 … dst = dst.batch(batch_size=s, input_columns=in_nms, output_columns=out_nms, per_batch_map=f,
393 def test_exceptions_config(gen_num, batch_size, in_cols, per_batch_map): argument
395 … per_batch_map=per_batch_map)
|
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3_darknet53/src/ |
| D | yolo_dataset.py | 172 … ds = ds.batch(batch_size, per_batch_map=multi_scale_trans, input_columns=dataset_column_names,
176 … ds = ds.batch(batch_size, per_batch_map=multi_scale_trans, input_columns=dataset_column_names,
|
| /third_party/mindspore/mindspore/dataset/engine/ |
| D | datasets.py | 429 def batch(self, batch_size, drop_remainder=False, num_parallel_workers=None, per_batch_map=None, argument
496 …return BatchDataset(self, batch_size, drop_remainder, num_parallel_workers, per_batch_map, input_c…
2164 …lf, input_dataset, batch_size, drop_remainder=False, num_parallel_workers=None, per_batch_map=None, argument
2180 self.per_batch_map = per_batch_map
2199 … self.output_columns, self.column_order, self.batch_size_func, self.per_batch_map,
2245 if self.per_batch_map is None:
2267 … initargs=([self.per_batch_map], arg_q_list, res_q_list))
2281 …self.per_batch_map = _PythonCallable(self.per_batch_map, idx, self.process_pool, arg_q_list, res_q…
2293 if self.per_batch_map is not None:
2294 self.per_batch_map = FuncWrapper(self.per_batch_map)
|
| D | validators.py | 601 [batch_size, drop_remainder, num_parallel_workers, per_batch_map,
620 if (pad_info is not None) and (per_batch_map is not None):
628 if (per_batch_map is None) != (input_columns is None):
634 if len(input_columns) != (len(ins.signature(per_batch_map).parameters) - 1):
|