Searched refs:gpu_archs (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| D | build_defs.bzl | 151 "--arch=%s" % ",".join(ctx.attr.gpu_archs),
186 "gpu_archs": attr.string_list(),
246 gpu_archs = [],
276 gpu_archs: The list of GPU architectures to compile for.
348 gpu_archs = gpu_archs,
430 gpu_archs = cuda_gpu_architectures() or rocm_gpu_architectures(),
|
| /external/tensorflow/third_party/nccl/ |
| D | build_defs.bzl.tpl | 79 if not ctx.attr.gpu_archs:
92 for arch in ctx.attr.gpu_archs:
152 "gpu_archs": attr.string_list(),
189 for code in ctx.attr.gpu_archs:
221 "gpu_archs": attr.string_list(),
361 gpu_archs = cuda_gpu_architectures(),
393 gpu_archs = cuda_gpu_architectures(),
|
| /external/clang/lib/Driver/ |
| D | Tools.cpp | 11206 std::vector<std::string> gpu_archs = in ConstructJob() local
11208 assert(gpu_archs.size() == 1 && "Exactly one GPU Arch required for ptxas."); in ConstructJob()
11209 const std::string& gpu_arch = gpu_archs[0]; in ConstructJob()
|