Searched defs:bucket_batch_sizes (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/ |
| D | bucket_batch_by_length_node.cc | 26 const std::vector<int32_t> &bucket_boundaries, const std::vector<int32_t> &bucket_batch_sizes, in BucketBatchByLengthNode()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | bucket_batch_by_length_op.cc | 29 const std::vector<int32_t> &bucket_batch_sizes, in BucketBatchByLengthOp()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/ir/datasetops/ |
| D | bindings.cc | 123 … const py::dict &pad_info, bool pad_to_bucket_boundary, bool drop_remainder) { in __anon10ea60a00a02()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/dataset/engine/ |
| D | datasets.py | 482 …def bucket_batch_by_length(self, column_names, bucket_boundaries, bucket_batch_sizes, element_leng… argument
2553 …def __init__(self, input_dataset, column_names, bucket_boundaries, bucket_batch_sizes, element_len… argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/api/ |
| D | datasets.cc | 432 const std::vector<int32_t> &bucket_boundaries, const std::vector<int32_t> &bucket_batch_sizes, in BucketBatchByLengthDataset()
|