Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_asm_compiler.cc71 std::vector<uint8> cubin(static_cast<uint8*>(cubin_out), in LinkGpuAsm() local
Dcuda_gpu_executor.cc216 port::Status GpuExecutor::LoadModuleFromCuBin(const char* cubin, in LoadModuleFromCuBin()
270 const char* cubin = spec.cuda_cubin_in_memory().bytes(); in GetKernel() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnvptx_compiler.cc392 std::vector<uint8> cubin = CompileGpuAsmOrGetCachedResult( in CompileTargetBinary() local
/external/tensorflow/tensorflow/stream_executor/gpu/
Dasm_compiler.cc335 std::string cubin; in CompileGpuAsm() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc382 port::Status GpuExecutor::LoadModuleFromCuBin(const char* cubin, in LoadModuleFromCuBin()