Home
last modified time | relevance | path

Searched refs:cubin_bytes (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dmodule_spec.h44 void AddCudaCubinInMemory(port::ArraySlice<const uint8> cubin_bytes) { in AddCudaCubinInMemory() argument
45 CHECK(!cubin_bytes.empty()); in AddCudaCubinInMemory()
47 cuda_cubin_in_memory_ = cubin_bytes; in AddCudaCubinInMemory()
Dkernel_spec.h341 MultiKernelLoaderSpec *AddCudaCubinInMemory(const char *cubin_bytes,
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_driver.h220 static port::Status LoadCubin(GpuContext* context, const char* cubin_bytes,
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_driver.cc573 const char* cubin_bytes, in LoadCubin() argument
576 CUresult result = cuModuleLoadFatBinary(module, cubin_bytes); in LoadCubin()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_driver.cc466 const char* cubin_bytes, in LoadCubin() argument