Searched refs:print_ptx (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ |
| D | tf_to_kernel.cc | 111 bool embed_memref_prints, bool print_ptx, bool enable_ftz, in Run() argument
123 /*generate_fatbin=*/true, print_ptx, enable_ftz, in Run()
152 llvm::cl::opt<bool> print_ptx( in main() local
180 embed_memref_prints, print_ptx, enable_ftz, cpu_codegen); in main()
|
| D | kernel_creator.cc | 395 bool generate_fatbin, bool print_ptx, in GenerateDeviceCode() argument
405 gpu_binary_attr_name, architectures, generate_fatbin, print_ptx, in GenerateDeviceCode()
433 bool embed_memref_prints, bool generate_fatbin, bool print_ptx, in GenerateKernelForTfCode() argument
456 print_ptx, enable_ftz)); in GenerateKernelForTfCode()
|
| D | kernel_creator.h | 43 bool print_ptx = false, bool enable_ftz = false, bool cpu_codegen = false);
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | gpu_kernel_to_blob_pass.cc | 56 bool generate_fatbin, bool print_ptx, bool enable_ftz) { in GpuKernelToBlobPass() argument
62 print_ptx_ = print_ptx; in GpuKernelToBlobPass()
256 bool generate_fatbin, bool print_ptx, bool enable_ftz) { in CreateGpuKernelToBlobPass() argument
258 blob_annotation, architectures, generate_fatbin, print_ptx, enable_ftz); in CreateGpuKernelToBlobPass()
|
| D | passes.h | 72 bool print_ptx = false, bool enable_ftz = false);
|