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
|