Searched defs:enable_ftz (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ |
| D | tf_to_kernel.cc | 112 bool print_ptx, bool print_llvmir, bool enable_ftz, in Run()
160 llvm::cl::opt<bool> enable_ftz( in main() local
|
| D | tf_framework_c_interface.cc | 146 bool enable_ftz, bool cpu_codegen) { in Compile()
233 int64_t max_supported_rank, bool enable_ftz, bool cpu_codegen) { in _mlir_ciface_tf_jit_compile()
|
| D | kernel_creator.cc | 465 bool print_ptx, bool print_llvmir, bool enable_ftz) { in GenerateDeviceCode()
519 bool print_llvmir, bool enable_ftz, bool cpu_codegen, bool jit_compile) { in GenerateKernelForTfCode()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | gpu_kernel_to_blob_pass.cc | 55 bool print_llvmir, bool enable_ftz) { in GpuKernelToBlobPass()
234 bool print_ptx, bool print_llvmir, bool enable_ftz) { in CreateGpuKernelToBlobPass()
|
| D | tf_to_jit_invocations.cc | 212 bool enable_ftz; member
260 bool enable_ftz, bool cpu_codegen) { in PopulateTFToJITInvocationPatterns()
270 int64_t max_supported_rank, bool enable_ftz, bool cpu_codegen) { in CreateTFToJITInvocationPass()
|
| D | tf_framework_legalize_to_llvm.cc | 331 Value enable_ftz = rewriter.create<LLVM::ConstantOp>( in matchAndRewrite() local
|