Searched defs:element_length_function (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 | 27 std::shared_ptr<TensorOp> element_length_function, in BucketBatchByLengthNode()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | bucket_batch_by_length_op.cc | 30 … std::shared_ptr<TensorOp> element_length_function, const PadInfo &pad_info, 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 …_by_length(self, column_names, bucket_boundaries, bucket_batch_sizes, element_length_function=None, argument
2553 …(self, input_dataset, column_names, bucket_boundaries, bucket_batch_sizes, element_length_function, argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/api/ |
| D | datasets.cc | 433 const std::function<MSTensorVec(MSTensorVec)> &element_length_function, in BucketBatchByLengthDataset()
|