Searched defs:step_num (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/include/backend/distributed/ps/ps_cache/ |
| D | ps_data_channel.h | 28 PsDataChannel(const std::string &channel_name, size_t step_num) in PsDataChannel()
41 void set_step_num(size_t step_num) { step_num_ = step_num; } in set_step_num()
|
| /third_party/mindspore/mindspore-src/source/tests/st/data_transfer/ |
| D | test_tdt_data_transfer.py | 54 def op_network_with_epoch(network, step_num): argument
115 def op_network_with_step_num(dataset, step_num): argument
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/dataset/ |
| D | test_callbacks.py | 81 def verify_events(events, epoch_num, step_num, step_size=1, map_num=1, repeat=1): argument
122 def generate_expected(epoch_num, step_num, step_size=1, map_num=1, repeat=1): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/ps/ps_cache/ps_data/ |
| D | ps_data_prefetch.cc | 30 void PsDataPrefetch::CreateDataChannel(const std::string &channel_name, size_t step_num) { in CreateDataChannel()
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/stub/ps/ |
| D | ps_core_stub.cc | 27 void PsDataPrefetch::CreateDataChannel(const std::string &channel_name, size_t step_num) {} in CreateDataChannel()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/train/callback/ |
| D | _summary_collector.py | 60 step_num = 'step_num' variable in LineageMetadata
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/perf/ |
| D | profiling.cc | 187 for (auto step_num = start_step; step_num <= end_step; step_num++) { in GetRecordEntryFieldValue() local
709 void ProfilingManager::RecordEndOfEpoch(uint32_t step_num) { in RecordEndOfEpoch()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ |
| D | context.py | 753 def _set_cur_step_num(self, step_num): argument
|