Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_gpu_executor.cc256 kernelname = &spec.cuda_cubin_in_memory().kernelname(); in GetKernel()
258 const char* hsaco = spec.cuda_cubin_in_memory().bytes(); in GetKernel()
378 reinterpret_cast<const char*>(spec.cuda_cubin_in_memory().data()), in LoadModule()
381 static_cast<const void*>(spec.cuda_cubin_in_memory().data()))); in LoadModule()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_gpu_executor.cc253 kernelname = &spec.cuda_cubin_in_memory().kernelname(); in GetKernel()
254 const char* cubin = spec.cuda_cubin_in_memory().bytes(); in GetKernel()
336 reinterpret_cast<const char*>(spec.cuda_cubin_in_memory().data()), in LoadModule()
339 static_cast<const void*>(spec.cuda_cubin_in_memory().data()))); in LoadModule()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dmodule_spec.h33 port::ArraySlice<const uint8> cuda_cubin_in_memory() const { // non-absl ok in cuda_cubin_in_memory() function
H A Dkernel_spec.h302 const CudaCubinInMemory &cuda_cubin_in_memory() const { in cuda_cubin_in_memory() function