Home
last modified time | relevance | path

Searched refs:gpu_arch (Results 1 – 4 of 4) sorted by relevance

/external/ComputeLibrary/src/core/CL/
DCLCompileContext.cpp243 GPUTarget gpu_arch = get_arch_from_target(_device.target()); in generate_build_options() local
245 static_cast<std::underlying_type<GPUTarget>::type>(gpu_arch)); in generate_build_options()
273 if(gpu_arch != GPUTarget::UNKNOWN && gpu_arch != GPUTarget::MIDGARD && get_ddk_version() >= 11) in generate_build_options()
/external/ComputeLibrary/tests/validation/fixtures/
DGEMMFixture.h235 DataType data_type, GPUTarget gpu_arch) in setup() argument
244 …ape, bias_shape, data_type, alpha, beta, broadcast_bias, fp16_mixed_precision, act_info, gpu_arch); in setup()
264 … bool fp16_mixed_precision, const ActivationLayerInfo &act_info, GPUTarget gpu_arch) in compute_target() argument
281 …gemm.configure(gpu_arch, lhs.info(), rhs.info(), bias.info(), dst.info(), alpha, beta, false, resh… in compute_target()
359 const ActivationLayerInfo &act_info, DataType data_type, GPUTarget gpu_arch) in setup() argument
371 …ape, rhs_shape, bias_shape, data_type, alpha, beta, m_h, fp16_mixed_precision, act_info, gpu_arch); in setup()
387 … bool fp16_mixed_precision, const ActivationLayerInfo &act_info, GPUTarget gpu_arch) in compute_target() argument
404 …gemm.configure(gpu_arch, lhs.info(), rhs.info(), bias.info(), dst.info(), alpha, beta, false, resh… in compute_target()
481 const ActivationLayerInfo &act_info, DataType data_type, GPUTarget gpu_arch) in setup() argument
504 …_info, rhs_info, data_type, alpha, beta, broadcast_bias, fp16_mixed_precision, act_info, gpu_arch); in setup()
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision.cc116 std::vector<std::string> gpu_arch = absl::StrSplit(gcnArchName, ":"); in HasFastFP16Support()
117 return !gpu_arch.empty() && FP16SupportedDevices.contains(gpu_arch[0]); in HasFastFP16Support()
/external/clang/lib/Driver/
DTools.cpp11209 const std::string& gpu_arch = gpu_archs[0]; in ConstructJob() local
11214 StringToCudaArch(gpu_arch)); in ConstructJob()
11258 CmdArgs.push_back(Args.MakeArgString(gpu_arch)); in ConstructJob()