Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/
Dasm_compiler.cc160 int cc_major; in CompileGpuAsm() local
220 static void LogPtxasTooOld(const std::string& ptxas_path, int cc_major, in LogPtxasTooOld()
248 port::StatusOr<std::vector<uint8>> CompileGpuAsm(int cc_major, int cc_minor, in CompileGpuAsm()
/external/tensorflow/tensorflow/core/util/autotune_maps/
Dautotune_maps_utils.cc43 int cc_major; in DeviceIdToIdentifierHelper() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnvptx_compiler.h111 int cc_major; member
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dgpu_kernel_to_blob_pass.cc152 int cc_major = arch / 10; in GetGpuBinaryBlob() local
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_gpu_executor.cc1008 int cc_major; in CreateDeviceDescription() local
Dcuda_driver.cc1367 /* static */ port::Status GpuDriver::GetComputeCapability(int* cc_major, in GetComputeCapability()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_driver.cc1002 /* static */ port::Status GpuDriver::GetComputeCapability(int* cc_major, in GetComputeCapability()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc1040 int64_t MinSystemMemory(int64_t available_memory, int cc_major) { in MinSystemMemory()