Searched refs:gpu_asm_opts (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | BUILD | 213 name = "gpu_asm_opts",
214 hdrs = ["gpu_asm_opts.h"],
238 ":gpu_asm_opts",
269 ":gpu_asm_opts",
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | gpu_kernel_to_blob_pass.cc | 165 auto gpu_asm_opts = xla::gpu::PtxOptsFromConfig(config); in GetGpuBinaryBlob() local
203 cc_major, cc_minor, ptx.c_str(), gpu_asm_opts)); in GetGpuBinaryBlob()
225 gpu_asm_opts.preferred_cuda_dir); in GetGpuBinaryBlob()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | BUILD | 689 "//tensorflow/stream_executor/gpu:gpu_asm_opts",
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | BUILD | 1622 "//tensorflow/stream_executor/gpu:gpu_asm_opts",
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 3799 "//tensorflow/stream_executor/gpu:gpu_asm_opts",
|