| /third_party/mindspore/tests/st/networks/models/resnet50/src_thor/ |
| D | dataset_helper.py | 64 def loop_size(self): member in DatasetHelper
66 return self.iter.loop_size
73 self.loop_size = 1
76 self.loop_size = dataset.get_dataset_size()
78 self.loop_size = dataset.__loop_size__
79 dataset.__transfer_dataset__ = _exec_datagraph(dataset, self.loop_size)
107 loop_size = dataset.__loop_size__
108 if dataset.get_dataset_size() % loop_size != 0:
111 loop_count = int(dataset.get_dataset_size() / loop_size)
120 loop_size = dataset.__loop_size__ + iter_first_order
[all …]
|
| D | model_thor.py | 437 loop_size = dataset_helper.loop_size()
451 cb_params.cur_step_num += loop_size
|
| /third_party/mindspore/mindspore/train/train_thor/ |
| D | dataset_helper.py | 133 loop_size = dataset.__loop_size__
134 … if loop_size <= dataset.get_dataset_size() and dataset.get_dataset_size() % loop_size != 0:
137 sink_count = math.ceil(dataset.get_dataset_size() / loop_size)
160 loop_size = dataset.__loop_size__ + iter_first_order
161 sink_count = int(sink_size / loop_size) * 2
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/ |
| D | hccl_kernel.cc | 179 ulong loop_size = hccl_data_type_list_.size(); in GetOutputSizeList() local
181 loop_size *= static_cast<ulong>(rank_size); in GetOutputSizeList()
184 loop_size = AnfAlgo::GetOutputTensorNum(anf_node); in GetOutputSizeList()
186 for (ulong i = 0; i < loop_size; ++i) { in GetOutputSizeList()
|
| /third_party/mindspore/mindspore/train/ |
| D | dataset_helper.py | 380 loop_size = dataset.__loop_size__
381 … if loop_size <= dataset.get_dataset_size() and dataset.get_dataset_size() % loop_size != 0:
384 sink_count = math.ceil(dataset.get_dataset_size() / loop_size)
|
| /third_party/skia/third_party/externals/spirv-tools/source/opt/ |
| D | loop_peeling.cpp | 592 CodeMetrics loop_size; in ProcessFunction() local
593 loop_size.Analyze(*loop); in ProcessFunction()
595 auto try_peel = [&loop_size, &modified, this](Loop* loop_to_peel) -> Loop* { in ProcessFunction()
603 ProcessLoop(loop_to_peel, &loop_size); in ProcessFunction()
625 CodeMetrics* loop_size) { in ProcessLoop() argument
737 if (factor * loop_size->roi_size_ > code_grow_threshold_) { in ProcessLoop()
740 loop_size->roi_size_ *= factor; in ProcessLoop()
|
| D | loop_peeling.h | 327 std::pair<bool, Loop*> ProcessLoop(Loop* loop, CodeMetrics* loop_size);
|
| /third_party/spirv-tools/source/opt/ |
| D | loop_peeling.cpp | 592 CodeMetrics loop_size; in ProcessFunction() local
593 loop_size.Analyze(*loop); in ProcessFunction()
595 auto try_peel = [&loop_size, &modified, this](Loop* loop_to_peel) -> Loop* { in ProcessFunction()
603 ProcessLoop(loop_to_peel, &loop_size); in ProcessFunction()
625 CodeMetrics* loop_size) { in ProcessLoop() argument
737 if (factor * loop_size->roi_size_ > code_grow_threshold_) { in ProcessLoop()
740 loop_size->roi_size_ *= factor; in ProcessLoop()
|
| D | loop_peeling.h | 327 std::pair<bool, Loop*> ProcessLoop(Loop* loop, CodeMetrics* loop_size);
|
| /third_party/skia/third_party/externals/swiftshader/third_party/SPIRV-Tools/source/opt/ |
| D | loop_peeling.cpp | 592 CodeMetrics loop_size; in ProcessFunction() local
593 loop_size.Analyze(*loop); in ProcessFunction()
595 auto try_peel = [&loop_size, &modified, this](Loop* loop_to_peel) -> Loop* { in ProcessFunction()
603 ProcessLoop(loop_to_peel, &loop_size); in ProcessFunction()
625 CodeMetrics* loop_size) { in ProcessLoop() argument
737 if (factor * loop_size->roi_size_ > code_grow_threshold_) { in ProcessLoop()
740 loop_size->roi_size_ *= factor; in ProcessLoop()
|
| D | loop_peeling.h | 327 std::pair<bool, Loop*> ProcessLoop(Loop* loop, CodeMetrics* loop_size);
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | resource.h | 85 int64_t loop_size() { return loop_size_; } in loop_size() function
|
| D | pipeline.cc | 161 int64_t loop_size = ConfigManager::GetInstance().iter_num(); in SetLoopCount() local
164 resource->set_vm_loop(!context_ptr->get_param<bool>(MS_CTX_IS_MULTI_GRAPH_SINK), loop_size); in SetLoopCount()
167 resource->set_vm_loop(!run_with_mind_rt, loop_size); in SetLoopCount()
169 …FO) << "Change vm_loop_flag to " << resource->vm_loop_flag() << ", set loop_size to " << loop_size; in SetLoopCount()
1076 int64_t loop_size = info_[phase]->resource->loop_size(); in Run() local
1079 vm_loop = loop_size; in Run()
1083 ConfigManager::GetInstance().set_gpu_loopsink_size(loop_size); in Run()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | config_manager.h | 58 int64_t loop_size() const { return loop_size_; } in loop_size() function
|