Searched refs:PadInfo (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | batch_op.h | 38 using PadInfo = std::map<std::string, std::pair<TensorShape, std::shared_ptr<Tensor>>>; variable
67 Builder &SetPaddingMap(const PadInfo &pad_map, bool pad = true) {
125 PadInfo builder_pad_map_;
155 py::function batch_size_func, py::function batch_map_func, PadInfo pad_map);
158 const std::vector<std::string> &, PadInfo pad_map);
207 static Status PadColumns(std::unique_ptr<TensorQTable> *table, const PadInfo &pad_info,
238 static Status UnpackPadInfo(const PadInfo &pad_info,
275 PadInfo pad_info_; // column names to perform padding on
|
| D | bucket_batch_by_length_op.h | 40 std::shared_ptr<TensorOp> element_length_function, const PadInfo &pad_info,
78 PadInfo pad_info_;
|
| D | bucket_batch_by_length_op.cc | 35 … std::shared_ptr<TensorOp> element_length_function, const PadInfo &pad_info, in BucketBatchByLengthOp()
133 PadInfo pad_info_copy = pad_info_; in PadAndBatchBucket()
|
| D | batch_op.cc | 54 py::function batch_size_func, py::function batch_map_func, PadInfo pad_map) in BatchOp()
78 const std::vector<std::string> &cols_to_map, PadInfo pad_map) in BatchOp()
427 Status BatchOp::PadColumns(std::unique_ptr<TensorQTable> *table, const PadInfo &pad_info, in PadColumns()
482 Status BatchOp::UnpackPadInfo(const PadInfo &pad_info, in UnpackPadInfo()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/ |
| D | bucket_batch_by_length_node.h | 71 …const std::map<std::string, std::pair<TensorShape, std::shared_ptr<Tensor>>> &PadInfo() const { re… in PadInfo() function
|
| /third_party/mindspore/tests/ut/cpp/dataset/common/ |
| D | common.h | 104 mindspore::dataset::PadInfo = {});
|
| D | common.cc | 89 … mindspore::dataset::PadInfo pad_map) { in Batch()
|
| /third_party/mindspore/tests/ut/cpp/dataset/ |
| D | batch_op_test.cc | 298 PadInfo m; in TEST_F()
|