Searched refs:gpu_dict (Results 1 – 2 of 2) sorted by relevance
10 def __init__(self, model_name, gpu_dict): argument11 if (model_name == None) or (gpu_dict == None):14 self._gpu = gpu_info.GPUInfo.FromDict(gpu_dict)
171 def __init__(self, model_name='', gpu_dict=None): argument172 if gpu_dict == None:173 gpu_dict = fake_gpu_info.FAKE_GPU_INFO174 super(FakeSystemInfo, self).__init__(model_name, gpu_dict)