Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_asm_compiler.cc65 std::vector<uint8> cubin(static_cast<uint8*>(cubin_out), in LinkGpuAsm() local
H A Dcuda_gpu_executor.cc200 port::Status GpuExecutor::LoadModuleFromCuBin(const char* cubin, in LoadModuleFromCuBin()
254 const char* cubin = spec.cuda_cubin_in_memory().bytes(); in GetKernel() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dnvptx_compiler.cc371 std::vector<uint8_t> cubin = CompileGpuAsmOrGetCachedResult( in CompileTargetBinary() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/
H A Dasm_compiler.cc332 std::string cubin; in CompileGpuAsm() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A Djit_utils.cpp1618 std::ofstream cubin(tmp_file_path, std::ios::out | std::ofstream::binary); in jit_pwise_function() local
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_gpu_executor.cc388 port::Status GpuExecutor::LoadModuleFromCuBin(const char* cubin, in LoadModuleFromCuBin()