Home
last modified time | relevance | path

Searched defs:gcn_arch_name (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/
Dgpu_backend_lib.cc579 std::vector<std::string> GetROCDLPaths(std::string gcn_arch_name, in GetROCDLPaths()
778 Status LinkROCDLIfNecessary(llvm::Module* module, std::string gcn_arch_name, in LinkROCDLIfNecessary()
799 std::string gcn_arch_name = compute_capability->gcn_arch_name(); in AMDGPUTargetModuleLinker() local
837 const std::string& gcn_arch_name) { in GetFeatureStrFromGCNArchName()
866 std::string gcn_arch_name = compute_capability->gcn_arch_name(); in AMDGPUGetTargetMachine() local
930 std::string gcn_arch_name = compute_capability->gcn_arch_name(); in CompileToHsaco() local
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Ddevice_description.h103 explicit RocmComputeCapability(const std::string &gcn_arch_name) in RocmComputeCapability()
108 std::string gcn_arch_name() { return gcn_arch_name_; } in gcn_arch_name() function
429 void set_rocm_compute_capability(std::string gcn_arch_name) { in set_rocm_compute_capability()
/external/pytorch/aten/src/ATen/cuda/tunable/
DTunable.cpp195 std::string gcn_arch_name = at::cuda::getCurrentDeviceProperties()->gcnArchName; in TuningResultsValidator() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc876 std::string gcn_arch_name; in CreateDeviceDescription() local
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc1914 std::string gcn_arch_name = in GetValidDeviceIds() local