Searched refs:enable_ftz (Results 1 – 6 of 6) 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()
156 llvm::cl::opt<bool> enable_ftz( in main() local
180 embed_memref_prints, print_ptx, enable_ftz, cpu_codegen); in main()
|
| D | kernel_creator.cc | 396 bool enable_ftz) { in GenerateDeviceCode() argument
406 enable_ftz)); in GenerateDeviceCode()
434 bool enable_ftz, bool cpu_codegen) { 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
63 enable_ftz_ = enable_ftz; 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);
|
| /external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| D | build_defs.bzl | 150 "--enable_ftz=%s" % (ctx.attr.data_type == "f32"),
|