Searched defs:print_ptx (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ |
| H A D | tf_to_kernel.cc | 111 bool embed_memref_prints, bool print_ptx, bool print_llvmir, in Run()
156 llvm::cl::opt<bool> print_ptx( in main() local
|
| H A D | kernel_creator.cc | 372 bool print_ptx, bool print_llvmir, bool enable_ftz, in GenerateDeviceCode()
428 int64_t max_supported_rank, bool embed_memref_prints, bool print_ptx, in GenerateKernelForTfCode()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| H A D | gpu_kernel_to_blob_pass.cc | 52 llvm::ArrayRef<std::string> architectures, bool print_ptx, in GpuKernelToBlobPass()
261 bool print_ptx, bool print_llvmir, bool enable_ftz) { in CreateGpuKernelToBlobPass()
|