Searched refs:out_col_names_ (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | batch_op.cc | 60 out_col_names_(out_col), in BatchOp()
310 size_t col_id = column_name_id_map_[out_col_names_[i]]; in MapColumns()
313 "Invalid data, column: " + out_col_names_[i] + in MapColumns()
395 CHECK_FAIL_RETURN_UNEXPECTED(ret_tuple.size() == out_col_names_.size(), in InvokeBatchMapFunc()
397 std::to_string(out_col_names_.size()) + in InvokeBatchMapFunc()
403 MS_LOG(INFO) << "column: " << out_col_names_[i] in InvokeBatchMapFunc()
533 …bool col_name_flag = (out_col_names_.empty() || out_col_names_ == in_col_names_); // true if col … in ComputeColMap()
537 if (out_col_names_.empty()) out_col_names_ = in_col_names_; in ComputeColMap()
549 …if (out_col_names_.size() == in_col_names_.size()) { // column names changed, but same number of … in ComputeColMap()
553 column_name_id_map_[out_col_names_[i]] = child_map_[in_col_names_[i]]; in ComputeColMap()
[all …]
|
| D | batch_op.h | 274 std::vector<std::string> out_col_names_; // output column name for per_batch_map variable
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/ |
| D | batch_node.cc | 42 out_col_names_(out_col_names), in BatchNode()
59 …:make_shared<BatchNode>(nullptr, batch_size_, drop_remainder_, pad_, in_col_names_, out_col_names_, in Copy()
106 … in_col_names_, out_col_names_, batch_size_func_, batch_map_func_, pad_map_); in Build()
165 args["output_columns"] = out_col_names_; in to_json()
|
| D | batch_node.h | 96 const std::vector<std::string> &OutColNames() const { return out_col_names_; } in OutColNames()
121 std::vector<std::string> out_col_names_; variable
|