Searched refs:CompileGpuAsm (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | asm_compiler.h | 37 port::StatusOr<std::vector<uint8>> CompileGpuAsm(int device_ordinal,
47 port::StatusOr<std::vector<uint8>> CompileGpuAsm(int cc_major, int cc_minor,
|
| D | asm_compiler.cc | 123 CompileGpuAsm(device_ordinal, ptx, compilation_options); in CompileGpuAsmOrGetCached()
141 port::StatusOr<std::vector<uint8>> CompileGpuAsm(int device_ordinal, in CompileGpuAsm() function
150 return CompileGpuAsm(cc_major, cc_minor, ptx_contents, options); in CompileGpuAsm()
198 port::StatusOr<std::vector<uint8>> CompileGpuAsm(int cc_major, int cc_minor, in CompileGpuAsm() function
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | gpu_kernel_to_blob_pass.cc | 202 tensorflow::se::CompileGpuAsm( in GetGpuBinaryBlob()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nvptx_compiler.cc | 376 StatusOr<std::vector<uint8>> maybe_cubin = se::CompileGpuAsm( in CompileGpuAsmOrGetCachedResult()
|