| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | pipeline_split.cc | 58 static int64_t InferStage(int64_t rank_id, int64_t stage_num, int64_t device_num) { in InferStage()
88 int64_t device_num = 0; in PipelineSplit() local
|
| /third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/ |
| D | generate_mindrecord.py | 12 device_num = 1 variable
|
| D | test_fl_fasterrcnn.py | 95 device_num = 1 variable
|
| /third_party/mindspore/tests/st/broadcast/ |
| D | lenet_broadcast_auto_parallel.py | 30 device_num = 2 variable
|
| /third_party/mindspore/tests/st/networks/models/bert/src/ |
| D | dataset.py | 26 def create_bert_dataset(epoch_size=1, device_num=1, rank=0, do_shuffle="true", data_dir=None, schem… argument
|
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3_darknet53/src/ |
| D | transforms.py | 240 def color_distortion(img, hue, sat, val, device_num): argument
371 anchors, num_classes, max_trial=10, device_num=1): argument
433 def preprocess_fn(image, box, config, input_size, device_num): argument
457 def __init__(self, config, device_num): argument
|
| D | yolo_dataset.py | 144 def create_yolo_dataset(image_dir, anno_path, batch_size, max_epoch, device_num, rank, argument
|
| /third_party/mindspore/tests/st/networks/models/bert/bert_performance/ |
| D | test_bert_thor.py | 108 def create_bert_dataset(device_num=1, rank=0, do_shuffle="true", data_dir=None, schema_dir=None): argument
143 def train_process_bert_thor(q, device_id, epoch_size, device_num): argument
|
| /third_party/mindspore/tests/st/networks/models/resnet50/ |
| D | test_resnet50_imagenet.py | 132 def train_process(q, device_id, epoch_size, device_num, enable_hccl): argument
230 def train_process_thor(q, device_id, epoch_size, device_num, enable_hccl): argument
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | launch_mul.cc | 63 auto device_num = parallel_context->device_num(); in ObtainMulInputsAddr() local
|
| /third_party/mindspore/tests/st/pynative/data_parallel/ |
| D | test_pynative_hccl_allreduce.py | 53 def train_allreduce_8p(q, device_id, device_num): argument
|
| D | test_pynative_hccl.py | 43 def train_allreduce_8p(q, device_id, device_num): argument
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel.h | 51 int64_t device_num = 1; member
|
| /third_party/mindspore/tests/st/networks/models/resnet50/src_thor/ |
| D | metric.py | 89 def __init__(self, batch_size, device_num): argument
|
| /third_party/mindspore/tests/st/networks/models/resnet50/src/ |
| D | metric.py | 89 def __init__(self, batch_size, device_num): argument
|
| /third_party/mindspore/tests/st/tbe_networks/ |
| D | hccl_2p_server_template.json | 8 "device_num": "2", string
|
| D | hccl_2p_evb_template.json | 8 "device_num": "2", string
|
| D | test_resnet_cifar_8p.py | 145 def train_process(q, device_id, epoch_size, num_classes, device_num, batch_size, enable_hccl): argument
|
| /third_party/mindspore/config/ |
| D | hccl_evb_multi_rank.json | 8 "device_num": "2", string
|
| D | hccl_single_machine_multi_rank.json | 7 "device_num": "8", string
97 "device_num": "number of devices used, the value is the nth power of 2", string
|
| /third_party/mindspore/tests/st/auto_parallel/ |
| D | optimizer_parallel.py | 277 device_num): argument
290 device_num): argument
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | batchnorm_info.cc | 224 ValuePtr device_num = MakeValue(forward_allreduce_group_[0].GetDevNum()); in InferReplaceOps() local
|
| D | uniform_real_info.cc | 149 int64_t device_num = stage_device_size_; in ReplaceNodeInputOrAttrs() local
|
| /third_party/mindspore/mindspore/parallel/ |
| D | _utils.py | 84 def _to_full_shapes(shapes, device_num): argument
|
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3/src/ |
| D | dataset.py | 295 def create_yolo_dataset(mindrecord_dir, batch_size=32, repeat_num=10, device_num=1, rank=0, argument
|