| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ps_data/ |
| D | ps_data_channel.h | 27 PsDataChannel(const std::string &channel_name, size_t step_num) in PsDataChannel() argument
29 step_num_(step_num), in PsDataChannel()
40 void set_step_num(size_t step_num) { step_num_ = step_num; } in set_step_num() argument
|
| D | ps_data_prefetch.cc | 25 void PsDataPrefetch::CreateDataChannel(const std::string &channel_name, size_t step_num) { in CreateDataChannel() argument
29 …"PS cache creates data channel(channel name:" << channel_name << ", step num:" << step_num << ")."; in CreateDataChannel()
35 channel->set_step_num(step_num); in CreateDataChannel()
37 auto channel = std::make_shared<PsDataChannel>(channel_name, step_num); in CreateDataChannel()
|
| D | ps_data_prefetch.h | 39 EXPORT void CreateDataChannel(const std::string &channel_name, size_t step_num);
|
| /third_party/mindspore/tests/st/data_transfer/ |
| D | test_tdt_data_transfer.py | 54 def op_network_with_epoch(network, step_num): argument
57 for _ in range(step_num):
115 def op_network_with_step_num(dataset, step_num): argument
126 return op_network_with_epoch(net_with_dataset, step_num)
143 iter_num = op_network_with_step_num(ds, step_num=beyond_step_num)
164 iter_num = op_network_with_step_num(ds, step_num=beyond_step_num)
|
| /third_party/mindspore/mindspore/profiler/parser/ |
| D | step_trace_parser.py | 474 for step_num in range(num_of_step):
476 'start': int(step_trace_info_all[iter_start][step_num].split(',')[start_time]),
477 'fp': int(step_trace_info_all[fp_start][step_num].split(',')[start_time]),
478 'bp': int(step_trace_info_all[bp_end][step_num].split(',')[end_time]),
479 'end': int(step_trace_info_all[iter_end][step_num].split(',')[end_time]),
487 cur_reduce_time = step_trace_info_all[reduce_idx][step_num]
|
| D | hccl_parser.py | 216 step_num = "1"
218 step_num = step_timestamps_info[step_timestamps_len - 1][0]
222 step_num = item[0]
223 return step_num
|
| D | integrator.py | 768 step_num = 1
777 … step_time_item = [str(step_num), tid, float(cur_step_start_time), cur_step_duration_time]
779 step_num += 1
1007 step_num = 1
1023 … step_time_item = [str(step_num), tid, cur_step_start_time, cur_step_duration_time]
1025 step_num += 1
1297 step_num = len(step_time_list)
1359 for step in range(step_num):
1367 metric.append(sum(metric) / step_num)
|
| /third_party/mindspore/tests/ut/data/profiler_data/profiler/ |
| D | step_trace_raw_6_detail_time.csv | 1 step_num,start_point,end_point,total,fp_point,bp_point,iteration_interval,fp_and_bp,tail,stream_24_…
|
| D | step_trace_raw_10_detail_time.csv | 1 step_num,start_point,end_point,total,fp_point,bp_point,iteration_interval,fp_and_bp,tail,stream_10_…
|
| D | step_trace_raw_0_detail_time.csv | 1 step_num,start_point,end_point,total,fp_point,bp_point,iteration_interval,fp_and_bp,tail,stream_10_…
|
| D | hccl_raw_6.csv | 1 step_num,communication_cost,wait_cost,link_info,communication_operator_cost
|
| /third_party/mindspore/tests/ut/python/dataset/ |
| D | test_callbacks.py | 82 def generate_expected(epoch_num, step_num, step_size=1, map_num=1, repeat=1): argument
87 event = [name, e + 1, s + 1, e * step_num * repeat + s + 1]
94 for s in range(step_num * repeat):
98 append("epoch_end", e, step_num * repeat - 1)
|
| /third_party/mindspore/mindspore/train/callback/ |
| D | _summary_collector.py | 55 step_num = 'step_num' variable in LineageMetadata
719 train_lineage[LineageMetadata.step_num] = cb_params.cur_step_num
|