Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/gpu/
Dasm_compiler.h38 const char* ptx_contents,
48 const char* ptx_contents,
Dasm_compiler.cc142 const char* ptx_contents, in CompileGpuAsm() argument
150 return CompileGpuAsm(cc_major, cc_minor, ptx_contents, options); in CompileGpuAsm()
199 const char* ptx_contents, in CompileGpuAsm() argument
213 tensorflow::WriteStringToFile(env, ptx_path, ptx_contents)); in CompileGpuAsm()
Dgpu_driver.h270 static port::Status LoadPtx(GpuContext* context, const char* ptx_contents,
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_driver.cc547 const char* ptx_contents, in LoadPtx() argument
551 GetDriverExecutor()->Schedule([context, ptx_contents, module, &ret, in LoadPtx()
554 void* ptx_data = const_cast<char*>(ptx_contents); in LoadPtx()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_driver.cc419 const char* ptx_contents, in LoadPtx() argument