Home
last modified time | relevance | path

Searched refs:LoadModuleFromCuBin (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc220 port::Status GpuExecutor::LoadModuleFromCuBin(const char* cubin, in LoadModuleFromCuBin() function in stream_executor::gpu::GpuExecutor
275 TF_RETURN_IF_ERROR(LoadModuleFromCuBin(cubin, &module)); in GetKernel()
355 TF_RETURN_IF_ERROR(LoadModuleFromCuBin( in LoadModule()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h271 port::Status LoadModuleFromCuBin(const char* cubin, GpuModuleHandle* module)
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc386 port::Status GpuExecutor::LoadModuleFromCuBin(const char* cubin, in LoadModuleFromCuBin() function in stream_executor::gpu::GpuExecutor