Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Dmodule_spec.h39 const char* cuda_ptx_in_memory() const { in cuda_ptx_in_memory() function
Dkernel_spec.h306 const CudaPtxInMemory &cuda_ptx_in_memory() const { in cuda_ptx_in_memory() function
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc278 kernelname = &spec.cuda_ptx_in_memory().kernelname(); in GetKernel()
284 const char* ptx = spec.cuda_ptx_in_memory().text(cc_major_, cc_minor_); in GetKernel()
286 ptx = spec.cuda_ptx_in_memory().default_text(); in GetKernel()
366 if (!spec.cuda_ptx_in_memory()) { in LoadModule()
372 LoadModuleFromPtx(spec.cuda_ptx_in_memory(), &cu_module)); in LoadModule()
374 const_cast<void*>(static_cast<const void*>(spec.cuda_ptx_in_memory()))); in LoadModule()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.cc274 module_spec.cuda_ptx_in_memory() == nullptr) { in ResolveConstantGlobals()