Searched defs:element_length_function (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/ |
| D | bucket_batch_by_length_node.cc | 34 std::shared_ptr<TensorOp> element_length_function, in BucketBatchByLengthNode()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | bucket_batch_by_length_op.cc | 35 … std::shared_ptr<TensorOp> element_length_function, const PadInfo &pad_info, in BucketBatchByLengthOp()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/ir/datasetops/ |
| D | bindings.cc | 124 bool drop_remainder) { in __anon4440814b0a02()
|
| /third_party/mindspore/mindspore/dataset/engine/ |
| D | datasets.py | 359 …_by_length(self, column_names, bucket_boundaries, bucket_batch_sizes, element_length_function=None, argument
2112 …(self, input_dataset, column_names, bucket_boundaries, bucket_batch_sizes, element_length_function, argument
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/ |
| D | datasets.cc | 409 std::function<MSTensorVec(MSTensorVec)> element_length_function, in BucketBatchByLengthDataset()
|