Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_gpu_executor.cc258 kernelname = &spec.cuda_ptx_in_memory().kernelname(); in GetKernel()
264 const char* ptx = spec.cuda_ptx_in_memory().text(cc_major_, cc_minor_); in GetKernel()
266 ptx = spec.cuda_ptx_in_memory().default_text(); in GetKernel()
346 if (!spec.cuda_ptx_in_memory()) { in LoadModule()
352 LoadModuleFromPtx(spec.cuda_ptx_in_memory(), &cu_module)); in LoadModule()
354 const_cast<void*>(static_cast<const void*>(spec.cuda_ptx_in_memory()))); in LoadModule()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dmodule_spec.h39 const char* cuda_ptx_in_memory() const { in cuda_ptx_in_memory() function
H A Dkernel_spec.h306 const CudaPtxInMemory &cuda_ptx_in_memory() const { in cuda_ptx_in_memory() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_executable.cc407 module_spec.cuda_ptx_in_memory() == nullptr)) { in ResolveConstantGlobals()