Searched defs:device_memory (Results 1 – 5 of 5) sorted by relevance
230 double device_memory = 1024.0 * 1024.0 * 1024.0 * 16.0; in TEST_F() local236 double device_memory = 1024.0 * 1024.0 * 1024.0 * 16.0; in TEST_F() local243 double device_memory = 1024.0 * 1024.0 * 1024.0 * 16.0; in TEST_F() local
29 auto device_memory = AscendMemoryPool::GetInstance().AllocTensorMem(size); in AllocDeviceMem() local
43 auto device_memory = GPUMemoryAllocator::GetInstance().AllocTensorMem(size); in AllocDeviceMem() local
195 Status PartitionForAllDevices(const size_t num_device, const double device_memory, in PartitionForAllDevices()268 Status DevicesMemoryControl(const size_t num_device, const double device_memory, const std::shared_… in DevicesMemoryControl()
1081 const auto device_memory = CostModelContext::GetInstance()->device_memory_capacity(); in ParallelStrategyRecSearch() local