Searched refs:gpus (Results 1 - 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/keras/_impl/keras/utils/
H A Dtraining_utils.py37 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 Dtraining_utils_test.py30 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 Dtest_environment.h45 VkPhysicalDevice gpus[16]; member in class:vk_testing::Environment
H A Dtest_environment.cpp121 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 Dnccl_manager_test.cc40 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 Dvulkaninfo.c2290 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 Dcommand_line.md261 on several gpus at workers and several parameter servers.

Completed in 233 milliseconds