Searched refs:in_col_names_ (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/ |
| D | batch_node.cc | 41 in_col_names_(in_col_names), in BatchNode()
59 …::make_shared<BatchNode>(nullptr, batch_size_, drop_remainder_, pad_, in_col_names_, out_col_names… in Copy()
87 if (batch_map_func_ && in_col_names_.empty()) { in ValidateParams()
106 … in_col_names_, out_col_names_, batch_size_func_, batch_map_func_, pad_map_); in Build()
112 in_col_names_, pad_map_)); in Build()
164 args["input_columns"] = in_col_names_; in to_json()
|
| D | batch_node.h | 95 const std::vector<std::string> &InColNames() const { return in_col_names_; } in InColNames()
120 std::vector<std::string> in_col_names_; variable
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | batch_op.cc | 59 in_col_names_(in_col), in BatchOp()
83 in_col_names_(cols_to_map), in BatchOp()
251 if (!in_col_names_.empty()) { in MakeBatchedRow()
287 TensorTable in_cols(in_col_names_.size(), TensorRow(num_rows, nullptr)), out_cols; in MapColumns()
290 for (size_t i = 0; i < in_col_names_.size(); i++) in_col_name_id.insert({in_col_names_[i], i}); in MapColumns()
518 …if (in_col_names_.empty()) { // if per_batch_map is not set, do not need to deal with out_col_nam… in ComputeColMap()
527 for (const auto &col : in_col_names_) { in ComputeColMap()
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()
544 for (const auto &col : in_col_names_) { in ComputeColMap()
[all …]
|
| D | batch_op.h | 273 const std::vector<std::string> in_col_names_; // input column name for per_batch_map variable
|