Home
last modified time | relevance | path

Searched defs:cc_major (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/gpu/
Dasm_compiler.cc163 int cc_major; in CompileGpuAsm() local
223 static void LogPtxasTooOld(const std::string& ptxas_path, int cc_major, in LogPtxasTooOld()
251 port::StatusOr<std::vector<uint8>> CompileGpuAsm(int cc_major, int cc_minor, in CompileGpuAsm()
/external/tensorflow/tensorflow/core/util/autotune_maps/
Dautotune_maps_utils.cc40 int cc_major; in DeviceIdToIdentifierHelper() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnvptx_compiler.h105 int cc_major; member
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dgpu_kernel_to_blob_pass.cc169 int cc_major = arch / 10; in GetGpuBinaryBlob() local
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc966 int cc_major; in CreateDeviceDescription() local
Dcuda_driver.cc1373 /* static */ port::Status GpuDriver::GetComputeCapability(int* cc_major, in GetComputeCapability()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_driver.cc1000 /* static */ port::Status GpuDriver::GetComputeCapability(int* cc_major, in GetComputeCapability()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc971 int64 MinSystemMemory(int64_t available_memory, int cc_major) { in MinSystemMemory()