Searched refs:DSCallback (Results 1 – 17 of 17) sorted by relevance
| /third_party/mindspore/mindspore/dataset/callback/ |
| D | ds_callback.py | 25 class DSCallback: class
98 if self.__class__.ds_begin != DSCallback.ds_begin:
102 if self.__class__.ds_epoch_begin != DSCallback.ds_epoch_begin:
105 if self.__class__.ds_epoch_end != DSCallback.ds_epoch_end:
109 if self.__class__.ds_step_begin != DSCallback.ds_step_begin:
112 if self.__class__.ds_step_end != DSCallback.ds_step_end:
122 class WaitedDSCallback(Callback, DSCallback):
|
| D | __init__.py | 16 from .ds_callback import DSCallback, WaitedDSCallback
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/callback/ |
| D | ds_callback.h | 30 class DSCallback {
34 explicit DSCallback(int32_t step_size = 1) : step_size_(step_size) {} in step_size_()
37 ~DSCallback() = default;
|
| D | callback_manager.h | 42 void AddCallbacks(std::vector<std::shared_ptr<DSCallback>> callbacks);
79 …std::vector<std::shared_ptr<DSCallback>> callbacks_; // list of callbacks the DatasetOp needs to…
|
| D | py_ds_callback.h | 33 class PyDSCallback : public DSCallback {
37 : DSCallback(step_size), in DSCallback() function
|
| D | callback_manager.cc | 25 void CallbackManager::AddCallbacks(std::vector<std::shared_ptr<DSCallback>> callbacks) { in AddCallbacks()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/ |
| D | map_node.h | 35 std::vector<std::shared_ptr<DSCallback>> callbacks = {});
89 const std::vector<std::shared_ptr<DSCallback>> &Callbacks() const { return callbacks_; } in Callbacks()
120 std::vector<std::shared_ptr<DSCallback>> callbacks_;
|
| D | map_node.cc | 38 std::vector<std::shared_ptr<DSCallback>> callbacks) in MapNode()
183 … [](std::shared_ptr<DSCallback> cb) -> int32_t { return cb != nullptr ? cb->step_size() : 0; }); in to_json()
|
| /third_party/mindspore/tests/ut/cpp/dataset/ |
| D | ir_callback_test.cc | 39 class TestCallback : public DSCallback {
42 : DSCallback(step_size), in TestCallback()
134 std::shared_ptr<DSCallback> cb1 = tst_cb; in TEST_F()
154 std::vector<std::shared_ptr<DSCallback>> cbs = {}; in TEST_F()
194 std::shared_ptr<DSCallback> cb1 = tst_cb; in TEST_F()
212 std::vector<std::shared_ptr<DSCallback>> cbs = {}; in TEST_F()
262 std::shared_ptr<DSCallback> cb1 = tst_cb; in TEST_F()
331 std::shared_ptr<DSCallback> cb1 = tst_cb; in TEST_F()
|
| /third_party/mindspore/tests/ut/python/dataset/ |
| D | test_callbacks.py | 22 from mindspore.dataset.callback import DSCallback, WaitedDSCallback
32 class BaseCallback(DSCallback):
338 class BadCB(DSCallback):
394 class BadCB(DSCallback):
461 class FlagCallback(DSCallback):
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/liteapi/include/ |
| D | datasets.h | 62 class DSCallback; variable
199 std::vector<std::shared_ptr<DSCallback>> callbacks = {}) {
230 std::vector<std::shared_ptr<DSCallback>> callbacks = {}) {
262 std::vector<std::shared_ptr<DSCallback>> callbacks = {}) {
424 std::vector<std::shared_ptr<DSCallback>> callbacks);
|
| /third_party/mindspore/mindspore/dataset/engine/ |
| D | __init__.py | 25 from ..callback import DSCallback, WaitedDSCallback
|
| D | validators.py | 687 from mindspore.dataset.callback import DSCallback
704 type_check_list(callbacks, (DSCallback,), "callbacks")
706 type_check(callbacks, (DSCallback,), "callbacks")
712 type_check(callbacks, (list, DSCallback), "callbacks")
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | dataset_op.h | 346 …void AddCallbacks(std::vector<std::shared_ptr<DSCallback>> callbacks) { callback_manager_.AddCallb… in AddCallbacks()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/include/dataset/ |
| D | datasets.h | 77 class DSCallback; variable
316 std::vector<std::shared_ptr<DSCallback>> callbacks = {}) {
348 std::vector<std::shared_ptr<DSCallback>> callbacks = {}) {
381 std::vector<std::shared_ptr<DSCallback>> callbacks = {}) {
710 std::vector<std::shared_ptr<DSCallback>> callbacks);
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/ir/datasetops/ |
| D | bindings.cc | 200 … std::vector<std::shared_ptr<DSCallback>>(py_callbacks.begin(), py_callbacks.end())); in __anonb78612491402()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/ |
| D | datasets.cc | 466 … const std::shared_ptr<DatasetCache> &cache, std::vector<std::shared_ptr<DSCallback>> callbacks) { in MapDataset()
|