Searched refs:in_cols (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | batch_op.cc | 287 TensorTable in_cols(in_col_names_.size(), TensorRow(num_rows, nullptr)), out_cols; in MapColumns() local
296 in_cols[col_itr->second][i] = std::move((*in_q_table)[i][itr.second]); in MapColumns()
307 RETURN_IF_NOT_OK(InvokeBatchMapFunc(&in_cols, &out_cols, table_pair->second)); in MapColumns()
|
| /third_party/mindspore/tests/ut/python/dataset/ |
| D | test_var_batch_map.py | 393 def test_exceptions_config(gen_num, batch_size, in_cols, per_batch_map): argument
394 …ta1 = ds.GeneratorDataset((lambda: gen(gen_num)), ["num"]).batch(batch_size, input_columns=in_cols,
|