Home
last modified time | relevance | path

Searched refs:nccl (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/nccl/
DBUILD2 # Wrap NVIDIA (https://github.com/NVIDIA/nccl) NCCL with tensorflow ops.
30 "@local_config_nccl//:nccl",
53 "@local_config_nccl//:nccl",
/external/tensorflow/tensorflow/contrib/cmake/
Dtf_core_kernels.cmake165 # temporarily disable nccl (nccl itself needs to be ported to windows first)
166 "${tensorflow_source_dir}/tensorflow/contrib/nccl/kernels/nccl_manager.cc"
167 "${tensorflow_source_dir}/tensorflow/contrib/nccl/kernels/nccl_ops.cc"
168 "${tensorflow_source_dir}/tensorflow/contrib/nccl/ops/nccl_ops.cc"
174 # temporarily disable nccl as it needs to be ported with gpu
175 "${tensorflow_source_dir}/tensorflow/contrib/nccl/kernels/nccl_manager.cc"
176 "${tensorflow_source_dir}/tensorflow/contrib/nccl/kernels/nccl_ops.cc"
177 "${tensorflow_source_dir}/tensorflow/contrib/nccl/ops/nccl_ops.cc"
DCMakeLists.txt82 …option(tensorflow_PATH_CUDA_LIB "Additional library search path for cudnn, nccl, culibos" /usr/loc…
88 option(tensorflow_NCCL_INCLUDE "nccl.h header install path" /usr/include/)
98 option(tensorflow_PATH_NCCL_LIB "Override PATH_CUDA_LIB for nccl" ${tensorflow_PATH_CUDA_LIB})
446 message("nccl: ${nccl_LIBRARY}")
/external/tensorflow/tensorflow/tools/ci_build/
DDockerfile.gpu35 ln -s /usr/include/nccl.h /usr/local/cuda/include/nccl.h
/external/tensorflow/tensorflow/
Dopensource_only.files171 tensorflow/third_party/nccl/archive.BUILD
172 tensorflow/third_party/nccl/LICENSE
173 tensorflow/third_party/nccl/system.BUILD.tpl
174 tensorflow/third_party/nccl/nccl_configure.bzl
175 tensorflow/third_party/nccl/build_defs.bzl.tpl
176 tensorflow/third_party/nccl/BUILD
Dworkspace.bzl6 load("//third_party:nccl/nccl_configure.bzl", "nccl_configure")
600 build_file = clean_dep("//third_party:nccl/archive.BUILD"),
602 strip_prefix = "nccl-f93fe9bfd94884cec2ba711897222e0df5569a53",
604 …"https://mirror.bazel.build/github.com/nvidia/nccl/archive/f93fe9bfd94884cec2ba711897222e0df5569a5…
605 … "https://github.com/nvidia/nccl/archive/f93fe9bfd94884cec2ba711897222e0df5569a53.tar.gz",
/external/tensorflow/
DCODEOWNERS4 /tensorflow/core/nccl/ @azaks2 @chsigg
DRELEASE.md318 …GPUs and NCCL requires upgrade to [NCCL 2.2](https://developer.nvidia.com/nccl). See updated insta…
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_param_resolver_local.cc687 bool nccl = cp->instance.type == REDUCTION_COLLECTIVE && in CompleteInstanceFromInitializedIRec() local
690 if (broadcast || nccl) { in CompleteInstanceFromInitializedIRec()
691 WaitForGroup(ir, cp, is_source, broadcast, nccl, in CompleteInstanceFromInitializedIRec()
/external/tensorflow/tensorflow/core/kernels/
DBUILD199 "@local_config_nccl//:nccl",
200 "//tensorflow/core/nccl:nccl_lib",
366 "@local_config_nccl//:nccl",
367 "//tensorflow/core/nccl:nccl_lib",
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
DBUILD322 "@local_config_nccl//:nccl",
/external/cldr/tools/java/org/unicode/cldr/util/data/transforms/
Dinternal_raw_IPA-old.txt134895 nccl %11075