Searched refs:ptx_contents (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | asm_compiler.h | 38 const char* ptx_contents,
48 const char* ptx_contents,
|
| D | asm_compiler.cc | 142 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()
|
| D | gpu_driver.h | 270 static port::Status LoadPtx(GpuContext* context, const char* ptx_contents,
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_driver.cc | 547 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/ |
| D | rocm_driver.cc | 419 const char* ptx_contents, in LoadPtx() argument
|