| /third_party/mindspore/tests/ut/python/dataset/ |
| D | test_bucket_batch_by_length.py | 58 bucket_batch_sizes = [1, 1, 1, 1]
67 … _ = dataset.bucket_batch_by_length(invalid_column_names, bucket_boundaries, bucket_batch_sizes)
71 … _ = dataset.bucket_batch_by_length(column_names, empty_bucket_boundaries, bucket_batch_sizes)
75 … _ = dataset.bucket_batch_by_length(column_names, invalid_bucket_boundaries, bucket_batch_sizes)
79 … _ = dataset.bucket_batch_by_length(column_names, zero_start_bucket_boundaries, bucket_batch_sizes)
83 … _ = dataset.bucket_batch_by_length(column_names, negative_bucket_boundaries, bucket_batch_sizes)
87 … _ = dataset.bucket_batch_by_length(column_names, decreasing_bucket_boundaries, bucket_batch_sizes)
91 … dataset.bucket_batch_by_length(column_names, non_increasing_bucket_boundaries, bucket_batch_sizes)
111 _ = dataset.bucket_batch_by_length(column_names, bucket_boundaries, bucket_batch_sizes,
116 _ = dataset.bucket_batch_by_length(column_names, bucket_boundaries, bucket_batch_sizes,
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/ |
| D | bucket_batch_by_length_node.cc | 33 const std::vector<int32_t> &bucket_boundaries, const std::vector<int32_t> &bucket_batch_sizes, in BucketBatchByLengthNode() argument
39 bucket_batch_sizes_(bucket_batch_sizes), in BucketBatchByLengthNode()
|
| D | bucket_batch_by_length_node.h | 35 … const std::vector<int32_t> &bucket_boundaries, const std::vector<int32_t> &bucket_batch_sizes,
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | bucket_batch_by_length_op.cc | 34 const std::vector<int32_t> &bucket_batch_sizes, in BucketBatchByLengthOp() argument
41 bucket_batch_sizes_(bucket_batch_sizes), in BucketBatchByLengthOp()
|
| D | bucket_batch_by_length_op.h | 39 … const std::vector<int32_t> &bucket_boundaries, const std::vector<int32_t> &bucket_batch_sizes,
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/ir/datasetops/ |
| D | bindings.cc | 122 … std::vector<int32_t> bucket_boundaries, std::vector<int32_t> bucket_batch_sizes, in __anon4440814b0a02()
129 … dataset, toStringVector(column_names), bucket_boundaries, bucket_batch_sizes, in __anon4440814b0a02()
|
| /third_party/mindspore/mindspore/dataset/engine/ |
| D | validators.py | 538 [column_names, bucket_boundaries, bucket_batch_sizes, element_length_function, pad_info,
543 … type_check_list([column_names, bucket_boundaries, bucket_batch_sizes], (list,), nreq_param_list)
574 if len(bucket_batch_sizes) != len(bucket_boundaries) + 1:
577 all_int = all(isinstance(item, int) for item in bucket_batch_sizes)
581 all_non_negative = all(item > 0 for item in bucket_batch_sizes)
|
| D | datasets.py | 359 …def bucket_batch_by_length(self, column_names, bucket_boundaries, bucket_batch_sizes, element_leng… argument
425 return BucketBatchByLengthDataset(self, column_names, bucket_boundaries, bucket_batch_sizes,
2112 …def __init__(self, input_dataset, column_names, bucket_boundaries, bucket_batch_sizes, element_len… argument
2118 self.bucket_batch_sizes = replace_none(bucket_batch_sizes, [])
2126 … self.bucket_batch_sizes, self.element_length_function, self.pad_info,
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/include/dataset/ |
| D | datasets.h | 226 const std::vector<int32_t> &bucket_batch_sizes,
231 shared_from_this(), VectorStringToChar(column_names), bucket_boundaries, bucket_batch_sizes,
648 const std::vector<int32_t> &bucket_boundaries, const std::vector<int32_t> &bucket_batch_sizes,
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/ |
| D | datasets.cc | 408 const std::vector<int32_t> &bucket_boundaries, const std::vector<int32_t> &bucket_batch_sizes, in BucketBatchByLengthDataset() argument
432 bucket_boundaries, bucket_batch_sizes, c_func, in BucketBatchByLengthDataset()
|