Searched defs:device_memory (Results 1 – 7 of 7) sorted by relevance
234 double device_memory = 1024.0 * 1024.0 * 1024.0 * 16.0; in TEST_F() local241 double device_memory = 1024.0 * 1024.0 * 1024.0 * 16.0; in TEST_F() local249 double device_memory = 1024.0 * 1024.0 * 1024.0 * 16.0; in TEST_F() local276 double device_memory = 1024.0 * 1024.0 * 1024.0 * 16.0; in TEST_F() local
343 void PartitionPipelineStages(double device_memory, const std::shared_ptr<Graph> &graph) { in PartitionPipelineStages()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()
44 auto device_memory = GPUMemoryAllocator::GetInstance().AllocTensorMem(size); in AllocDeviceMem() local
94 auto device_memory = AscendMemoryPool::GetInstance().AllocTensorMem(size, false, stream_id_); in AllocDeviceMem() local
160 struct tu_device_memory *device_memory) in tu_rmv_log_heap_create()
1488 const auto device_memory = CostModelContext::GetInstance()->device_memory_capacity(); in ParallelStrategyRecSearch() local
990 auto device_memory = memory_manager_->MallocDeviceMemory("RefData input", ref_data_total_size); in InitRefDataDeviceTensor() local