Searched refs:all_devices (Results 1 – 4 of 4) sorted by relevance
412 atmi_device_t *all_devices = reinterpret_cast<atmi_device_t *>( in init_compute_and_memory() local436 g_atmi_machine.devices_by_type[ATMI_DEVTYPE_CPU] = &all_devices[cpus_begin]; in init_compute_and_memory()437 g_atmi_machine.devices_by_type[ATMI_DEVTYPE_GPU] = &all_devices[gpus_begin]; in init_compute_and_memory()438 g_atmi_machine.devices_by_type[ATMI_DEVTYPE_iGPU] = &all_devices[gpus_begin]; in init_compute_and_memory()439 g_atmi_machine.devices_by_type[ATMI_DEVTYPE_dGPU] = &all_devices[gpus_begin]; in init_compute_and_memory()442 all_devices[i].type = cpu_procs[proc_index].type(); in init_compute_and_memory()464 all_devices[i].type = gpu_procs[proc_index].type(); in init_compute_and_memory()
100 std::vector<std::unique_ptr<Device>> all_devices; in SetUp() local102 ->AddDevices(SessionOptions(), "", &all_devices)); in SetUp()103 for (std::unique_ptr<Device>& d : all_devices) { in SetUp()
182 def all_devices(): function325 devices = devices or all_devices()
213 DeviceMgr const* all_devices = device_mgr_; in InitializeDeviceAndFlr() local215 all_devices = parent_->remote_device_mgr(); in InitializeDeviceAndFlr()220 for (auto d : all_devices->ListDevices()) { in InitializeDeviceAndFlr()