Home
last modified time | relevance | path

Searched refs:gpus (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/python/keras/utils/
Dmulti_gpu_utils_test.py27 def check_if_compatible_devices(gpus=2): argument
32 if '/gpu:%d' % (gpus - 1) not in available_devices:
40 gpus = 2
48 if not check_if_compatible_devices(gpus=gpus):
60 parallel_model = keras.utils.multi_gpu_model(model, gpus=gpus)
63 parallel_model = keras.utils.multi_gpu_model(model, gpus=target_gpu_id)
68 gpus = 2
78 if not check_if_compatible_devices(gpus=gpus):
96 parallel_model = keras.utils.multi_gpu_model(model, gpus=gpus)
100 parallel_model = keras.utils.multi_gpu_model(model, gpus=target_gpu_id)
[all …]
Dmulti_gpu_utils.py37 def multi_gpu_model(model, gpus, cpu_merge=True, cpu_relocation=False): argument
156 if isinstance(gpus, (list, tuple)):
157 if len(gpus) <= 1:
160 'Received: `gpus=%s`' % gpus)
161 num_gpus = len(gpus)
162 target_gpu_ids = gpus
164 if gpus <= 1:
167 'Received: `gpus=%s`' % gpus)
168 num_gpus = gpus
181 'Try reducing `gpus`.' % (gpus, target_devices,
/external/tensorflow/tensorflow/
Dopensource_only.files73 tensorflow/third_party/gpus/BUILD
74 tensorflow/third_party/gpus/crosstool/clang/bin/crosstool_wrapper_driver_rocm.tpl
75 tensorflow/third_party/gpus/crosstool/clang/bin/crosstool_wrapper_driver_is_not_gcc.tpl
76 tensorflow/third_party/gpus/crosstool/CROSSTOOL.tpl
77 tensorflow/third_party/gpus/crosstool/CROSSTOOL_hipcc.tpl
78 tensorflow/third_party/gpus/crosstool/LICENSE
79 tensorflow/third_party/gpus/crosstool/windows/msvc_wrapper_for_nvcc.py.tpl
80 tensorflow/third_party/gpus/crosstool/BUILD.tpl
81 tensorflow/third_party/gpus/crosstool/BUILD
82 tensorflow/third_party/gpus/cuda/LICENSE
[all …]
Dworkspace.bzl3 load("//third_party/gpus:cuda_configure.bzl", "cuda_configure")
4 load("//third_party/gpus:rocm_configure.bzl", "rocm_configure")
/external/vulkan-validation-layers/tests/
Dtest_environment.cpp121 ASSERT_LE(count, ARRAY_SIZE(gpus)); in SetUp()
122 err = vkEnumeratePhysicalDevices(inst, &count, gpus); in SetUp()
126 vk_testing::PhysicalDevice phys_dev(gpus[0]); in SetUp()
142 devs_.push_back(new Device(gpus[i])); in SetUp()
Dtest_environment.h45 VkPhysicalDevice gpus[16]; variable
/external/tensorflow/tensorflow/contrib/cmake/
Dtf_stream_executor.cmake38 # "//third_party/gpus/cuda:cublas",
39 # "//third_party/gpus/cuda:cudnn",
47 # "//third_party/gpus/cuda:cuda_headers",
DCMakeLists.txt520 FILE(WRITE ${tensorflow_source_dir}/third_party/gpus/cuda/cuda_config.h
543 DESTINATION ${tensorflow_source_dir}/third_party/gpus/cuda/include
546 include_directories(${tensorflow_source_dir}/third_party/gpus)
/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager_test.cc40 std::vector<std::unique_ptr<BaseGPUDevice>> gpus; in GetGPUDevices() local
45 gpus.emplace_back(static_cast<BaseGPUDevice*>(device.release())); in GetGPUDevices()
48 return gpus; in GetGPUDevices()
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.keras.utils.pbtxt57 …argspec: "args=[\'model\', \'gpus\', \'cpu_merge\', \'cpu_relocation\'], varargs=None, keywords=No…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.keras.utils.pbtxt57 …argspec: "args=[\'model\', \'gpus\', \'cpu_merge\', \'cpu_relocation\'], varargs=None, keywords=No…
/external/tensorflow/tensorflow/lite/g3doc/models/segmentation/
Doverview.md41 …<li><a href="https://medium.com/tensorflow/tensorflow-lite-now-faster-with-mobile-gpus-developer-p…
/external/tensorflow/tensorflow/core/profiler/g3doc/
Dcommand_line.md261 on several gpus at workers and several parameter servers.
/external/llvm/docs/
DCompileCudaWithLLVM.rst118 your GPU <https://developer.nvidia.com/cuda-gpus>`_. For example, if you want
/external/swiftshader/third_party/llvm-7.0/llvm/docs/
DCompileCudaWithLLVM.rst79 <https://developer.nvidia.com/cuda-gpus>`_ of your GPU. For example, if you