Searched defs:num_device (Results 1 – 4 of 4) sorted by relevance
360 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()
782 size_t num_device = LongToSize(g_device_manager->stage_device_num()); in GenGatherStra() local806 int64_t num_device = g_device_manager->stage_device_num(); in GatherForDynamicShape() local858 size_t num_device = LongToSize(g_device_manager->stage_device_num()); in PrepareGather() local2567 size_t num_device = LongToSize(g_device_manager->stage_device_num()); in GenerateStrategyV1() local2626 static size_t CalMatmulBatchDimFactor(size_t num_device, const StrategyRec &str) { in CalMatmulBatchDimFactor()
254 int64_t num_device = stage; in Init() local
1487 size_t num_device = g_device_manager->DeviceNum(); in ParallelStrategyRecSearch() local