Home
last modified time | relevance | path

Searched defs:num_device (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/
Drec_partition.cc360 Status PartitionForAllDevices(size_t num_device, double device_memory, const std::shared_ptr<Graph>… in PartitionForAllDevices()
456 void DevicesMemoryControl(const size_t num_device, const double device_memory, const std::shared_pt… in DevicesMemoryControl()
Drec_generate_strategy.cc782 size_t num_device = LongToSize(g_device_manager->stage_device_num()); in GenGatherStra() local
806 int64_t num_device = g_device_manager->stage_device_num(); in GatherForDynamicShape() local
858 size_t num_device = LongToSize(g_device_manager->stage_device_num()); in PrepareGather() local
2567 size_t num_device = LongToSize(g_device_manager->stage_device_num()); in GenerateStrategyV1() local
2626 static size_t CalMatmulBatchDimFactor(size_t num_device, const StrategyRec &str) { in CalMatmulBatchDimFactor()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/
Ddevice_manager.cc254 int64_t num_device = stage; in Init() local
Dstep_auto_parallel.cc1487 size_t num_device = g_device_manager->DeviceNum(); in ParallelStrategyRecSearch() local