| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/ |
| D | root_node.h | 58 int32_t num_epochs() { return num_epochs_; } in num_epochs() function
61 void SetNumEpochs(int32_t num_epochs) { num_epochs_ = num_epochs; } in SetNumEpochs()
|
| D | epoch_ctrl_node.h | 37 explicit EpochCtrlNode(int32_t num_epochs) : RepeatNode() { repeat_count_ = num_epochs; } in EpochCtrlNode()
|
| D | epoch_ctrl_node.cc | 31 EpochCtrlNode::EpochCtrlNode(std::shared_ptr<DatasetNode> child, int32_t num_epochs) : RepeatNode()… in EpochCtrlNode()
|
| D | dataset_node.h | 311 void SetNumEpochs(int32_t num_epochs) { num_epochs_ = num_epochs; } in SetNumEpochs()
|
| /third_party/mindspore/tests/ut/cpp/dataset/ |
| D | c_api_epoch_ctrl_test.cc | 30 int32_t num_epochs = 2 + std::rand() % 3; in TEST_F() local
78 int32_t num_epochs = 1 + std::rand() % 4; in TEST_F() local
125 int32_t num_epochs = 2 + std::rand() % 5; in TEST_F() local
174 int32_t num_epochs = 1 + std::rand() % 5; in TEST_F() local
|
| D | ir_callback_test.cc | 234 size_t num_epochs = 2; in TEST_F() local
304 size_t num_epochs = 2; in TEST_F() local
|
| D | c_api_dataset_iterator_test.cc | 201 int32_t num_epochs = 3; in TEST_F() local
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/consumers/ |
| D | python_tree_consumer.h | 34 explicit PythonIteratorConsumer(int32_t num_epochs = -1) : IteratorConsumer(num_epochs) {} in IteratorConsumer() argument
|
| /third_party/mindspore/tests/mindspore_test_framework/apps/ |
| D | test_lamb_check_loss.py | 26 num_epochs = 1000 variable
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/pre/ |
| D | epoch_ctrl_pass.h | 75 int32_t num_epochs() { return num_epochs_; } in num_epochs() function
|
| D | epoch_ctrl_pass.cc | 88 int32_t num_epochs = finder.num_epochs(); in RunOnTree() local
|
| /third_party/mindspore/tests/ |
| D | dataset_mock.py | 58 def create_tuple_iterator(self, num_epochs=-1, do_copy=True): argument
61 def send(self, num_epochs=-1): argument
|
| /third_party/mindspore/mindspore/dataset/engine/ |
| D | iterators.py | 66 def __init__(self, dataset, num_epochs=-1, output_numpy=False, do_copy=True): argument
193 def __init__(self, dataset, columns=None, num_epochs=-1, output_numpy=False, do_copy=True): argument
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_bias_add.py | 64 def create_tuple_iterator(self, num_epochs=-1, do_copy=True): argument
|
| D | test_control_flow.py | 54 def create_tuple_iterator(self, num_epochs=1, do_copy=True): argument
|
| D | test_pipeline_shared_parameter.py | 55 def create_tuple_iterator(self, num_epochs=1, do_copy=True): argument
|
| D | test_auto_parallel_double_subgraphs.py | 154 def create_tuple_iterator(self, num_epochs=-1, do_copy=True): argument
|
| D | test_pipeline_split.py | 55 def create_tuple_iterator(self, num_epochs=1, do_copy=True): argument
|
| D | test_gather_v2_primitive.py | 62 def create_tuple_iterator(self, num_epochs=-1, do_copy=True): argument
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/ |
| D | iterator.cc | 78 Status Iterator::BuildAndLaunchTree(std::shared_ptr<Dataset> ds, int32_t num_epochs) { in BuildAndLaunchTree()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ |
| D | tree_adapter.cc | 175 Status TreeAdapter::Compile(std::shared_ptr<DatasetNode> input_ir, int32_t num_epochs) { in Compile()
|
| /third_party/mindspore/tests/mindspore_test_framework/utils/ |
| D | model_util.py | 108 def train(self, num_epochs, train_dataset): argument
|
| /third_party/mindspore/tests/st/hcom/ |
| D | hcom_sparsetensor.py | 79 def create_tuple_iterator(self, num_epochs=-1, do_copy=False): argument
|
| /third_party/mindspore/tests/st/mix_precision/ |
| D | utils.py | 110 def create_tuple_iterator(self, num_epochs=-1, do_copy=True): argument
|
| /third_party/mindspore/tests/st/pynative/loss_scale/ |
| D | test_loss_scale.py | 63 def create_tuple_iterator(self, num_epochs=-1, do_copy=True): argument
|