Home
last modified time | relevance | path

Searched defs:enable_ftz (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/
Dtf_to_kernel.cc111 bool embed_memref_prints, bool print_ptx, bool enable_ftz, in Run()
156 llvm::cl::opt<bool> enable_ftz( in main() local
Dkernel_creator.cc396 bool enable_ftz) { in GenerateDeviceCode()
434 bool enable_ftz, bool cpu_codegen) { in GenerateKernelForTfCode()
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dgpu_kernel_to_blob_pass.cc56 bool generate_fatbin, bool print_ptx, bool enable_ftz) { in GpuKernelToBlobPass()
256 bool generate_fatbin, bool print_ptx, bool enable_ftz) { in CreateGpuKernelToBlobPass()