Searched refs:gpus (Results 1 - 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/python/keras/_impl/keras/utils/ |
H A D | training_utils.py | 37 def multi_gpu_model(model, gpus): 48 E.g. if your `batch_size` is 64 and you use `gpus=2`, 62 gpus: Integer >= 2, number of on GPUs on which to create 94 parallel_model = multi_gpu_model(model, gpus=8) 111 ValueError: if the `gpus` argument does not match available devices. 117 if isinstance(gpus, (list, tuple)): 118 if len(gpus) <= 1: 120 'call `multi_gpu_model` with `len(gpus) >= 2`. ' 121 'Received: `gpus=%s`' % gpus) [all...] |
H A D | training_utils_test.py | 30 gpus = 2 47 parallel_model = keras.utils.multi_gpu_model(model, gpus=gpus) 51 parallel_model = keras.utils.multi_gpu_model(model, gpus=target_gpu_id) 56 gpus = 2 81 parallel_model = keras.utils.multi_gpu_model(model, gpus=gpus) 85 parallel_model = keras.utils.multi_gpu_model(model, gpus=target_gpu_id) 103 model, gpus=len(keras.backend._get_available_gpus()) + 1) 108 model, gpus [all...] |
/external/vulkan-validation-layers/tests/ |
H A D | test_environment.h | 45 VkPhysicalDevice gpus[16]; member in class:vk_testing::Environment
|
H A D | test_environment.cpp | 121 ASSERT_LE(count, ARRAY_SIZE(gpus)); 122 err = vkEnumeratePhysicalDevices(inst, &count, gpus); 126 vk_testing::PhysicalDevice phys_dev(gpus[0]); 142 devs_.push_back(new Device(gpus[i]));
|
/external/tensorflow/tensorflow/contrib/nccl/kernels/ |
H A D | nccl_manager_test.cc | 40 std::vector<BaseGPUDevice*> gpus; local 43 gpus.push_back(static_cast<BaseGPUDevice*>(d)); 48 return gpus; 57 LOG(ERROR) << "Running test with " << devices->size() << " gpus";
|
/external/vulkan-validation-layers/demos/ |
H A D | vulkaninfo.c | 2290 struct AppGpu *gpus; local 2343 gpus = malloc(sizeof(gpus[0]) * gpu_count); 2344 if (!gpus) ERR_EXIT(VK_ERROR_OUT_OF_HOST_MEMORY); 2346 AppGpuInit(&gpus[i], &inst, i, objs[i]); 2394 fprintf(out, "GPU id: <div class='val'>%u</div> (%s)</summary></details>\n", j, gpus[j].props.deviceName); 2396 printf("\t\tGPU id : %u (%s)\n", j, gpus[j].props.deviceName); 2400 AppGetPhysicalDeviceLayerExtensions(&gpus[j], layer_name, &count, &props); 2454 gpus[i].props.deviceName); 2458 printf("GPU id : %u (%s)\n", i, gpus[ [all...] |
/external/tensorflow/tensorflow/core/profiler/g3doc/ |
H A D | command_line.md | 261 on several gpus at workers and several parameter servers.
|
Completed in 233 milliseconds