Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.h75 const std::vector<uint8>& cubin() const { return cubin_; } in cubin() function
Dnvptx_compiler.cc605 string cubin; in CompilePtx() local
763 const std::vector<uint8> cubin = in RunBackend() local
Dgpu_executable.cc51 const string& ptx, const std::vector<uint8>& cubin, in GpuExecutable()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc220 bool GpuExecutor::LoadModuleFromCuBin(const char* cubin, CUmodule* module) { in LoadModuleFromCuBin()
278 const char *cubin = spec.cuda_cubin_in_memory().bytes(); in GetKernel() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc385 bool GpuExecutor::LoadModuleFromCuBin(const char* cubin, hipModule_t* module) { in LoadModuleFromCuBin()