Searched refs:reset_every_epoch (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/ |
| D | shuffle_node.cc | 30 …Node::ShuffleNode(std::shared_ptr<DatasetNode> child, int32_t shuffle_size, bool reset_every_epoch) in ShuffleNode() argument
31 : shuffle_size_(shuffle_size), shuffle_seed_(GetSeed()), reset_every_epoch_(reset_every_epoch) { in ShuffleNode()
79 bool reset_every_epoch = json_obj["reset_each_epoch"]; in from_json() local
80 *result = std::make_shared<ShuffleNode>(ds, buffer_size, reset_every_epoch); in from_json()
|
| D | shuffle_node.h | 33 ShuffleNode(std::shared_ptr<DatasetNode> child, int32_t shuffle_size, bool reset_every_epoch);
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | shuffle_op.cc | 41 …eOp(int32_t shuffle_size, uint32_t shuffle_seed, int32_t op_connector_size, bool reset_every_epoch) in ShuffleOp() argument
45 reshuffle_each_epoch_(reset_every_epoch), in ShuffleOp()
|
| D | shuffle_op.h | 58 …Op(int32_t shuffle_size, uint32_t shuffle_seed, int32_t op_connector_size, bool reset_every_epoch);
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/ir/datasetops/ |
| D | bindings.cc | 239 ….def(py::init([](std::shared_ptr<DatasetNode> self, int32_t shuffle_size, bool reset_every_epoch) { in __anonb78612491c02() argument
240 … auto shuffle = std::make_shared<ShuffleNode>(self, shuffle_size, reset_every_epoch); in __anonb78612491c02()
|