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.cc49 std::vector<uint8> cubin(static_cast<uint8*>(cubin_out), in LinkGpuAsm() local
Dcuda_gpu_executor.cc220 port::Status GpuExecutor::LoadModuleFromCuBin(const char* cubin, in LoadModuleFromCuBin()
274 const char* cubin = spec.cuda_cubin_in_memory().bytes(); in GetKernel() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnvptx_compiler.cc333 std::vector<uint8> cubin = CompileGpuAsmOrGetCachedResult( in CompileTargetBinary() local
/external/tensorflow/tensorflow/stream_executor/gpu/
Dasm_compiler.cc278 std::string cubin; in CompileGpuAsm() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc386 port::Status GpuExecutor::LoadModuleFromCuBin(const char* cubin, in LoadModuleFromCuBin()