Home
last modified time | relevance | path

Searched refs:cuda_cubin_in_memory (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dmodule_spec.h33 port::ArraySlice<const uint8> cuda_cubin_in_memory() const { in cuda_cubin_in_memory() function
Dkernel_spec.h302 const CudaCubinInMemory &cuda_cubin_in_memory() const { in cuda_cubin_in_memory() function
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc237 kernelname = &spec.cuda_cubin_in_memory().kernelname(); in GetKernel()
239 const char* hsaco = spec.cuda_cubin_in_memory().bytes(); in GetKernel()
372 reinterpret_cast<const char*>(spec.cuda_cubin_in_memory().data()), in LoadModule()
377 static_cast<const void*>(spec.cuda_cubin_in_memory().data()))); in LoadModule()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc277 kernelname = &spec.cuda_cubin_in_memory().kernelname(); in GetKernel()
278 const char *cubin = spec.cuda_cubin_in_memory().bytes(); in GetKernel()
368 reinterpret_cast<const char *>(spec.cuda_cubin_in_memory().data()), in LoadModule()
373 static_cast<const void *>(spec.cuda_cubin_in_memory().data()))); in LoadModule()