Searched defs:enable_ftz (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ |
| D | kernel_creator.cc | 113 int64_t max_supported_rank, bool enable_ftz, in LowerTFToJITInvocation()
142 int64_t max_supported_rank, bool enable_ftz, in LowerTFtoLoops()
372 bool print_ptx, bool print_llvmir, bool enable_ftz, in GenerateDeviceCode()
429 bool print_llvmir, bool enable_ftz, bool index_64bit, bool jit_compile, in GenerateKernelForTfCode()
|
| D | tf_to_kernel.cc | 112 bool enable_ftz, bool index_64bit, bool jit_compile, in Run()
163 llvm::cl::opt<bool> enable_ftz( in main() local
|
| D | tf_framework_c_interface.cc | 154 bool enable_ftz, bool index_64bit) { in Compile()
243 int64_t* unroll_factors_ptr, int64_t max_supported_rank, bool enable_ftz, in _mlir_ciface_tf_jit_compile()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | tf_to_jit_invocations.cc | 246 bool enable_ftz; member
293 int64_t max_supported_rank, bool enable_ftz, bool index_64bit, in PopulateTFToJITInvocationPatterns()
310 int64_t max_supported_rank, bool enable_ftz, bool index_64bit, in CreateTFToJITInvocationPass()
|
| D | gpu_kernel_to_blob_pass.cc | 53 bool print_llvmir, bool enable_ftz) { in GpuKernelToBlobPass()
261 bool print_ptx, bool print_llvmir, bool enable_ftz) { in CreateGpuKernelToBlobPass()
|
| D | tf_framework_legalize_to_llvm.cc | 278 Value enable_ftz = rewriter.create<LLVM::ConstantOp>( in matchAndRewrite() local
|