Searched refs:cpu_codegen (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| H A D | tf_to_jit_invocations.cc | 192 bool cpu_codegen) in PackJITCompileOpPattern()
199 cpu_codegen(cpu_codegen) {} in PackJITCompileOpPattern()
237 rewriter.getBoolAttr(cpu_codegen)); in matchAndRewrite()
248 bool cpu_codegen; member
263 bool index_64bit, bool cpu_codegen, in TFToJITInvocationPass()
270 cpu_codegen_ = cpu_codegen; in TFToJITInvocationPass()
294 bool cpu_codegen, bool jit_i64_indexed_for_large_tensors) { in PopulateTFToJITInvocationPatterns() argument
305 index_64bit_if_jit_compiling, cpu_codegen); in PopulateTFToJITInvocationPatterns()
311 bool cpu_codegen, bool jit_i64_indexed_for_large_tensors) { in CreateTFToJITInvocationPass() argument
314 cpu_codegen, jit_i64_indexed_for_large_tensors); in CreateTFToJITInvocationPass()
|
| H A D | rewriters.h | 60 bool cpu_codegen, bool jit_i64_indexed_for_large_tensors);
|
| H A D | passes.h | 55 bool enable_ftz = false, bool index_64bit = false, bool cpu_codegen = false,
|
| H A D | tf_framework_legalize_to_llvm.cc | 282 Value cpu_codegen = rewriter.create<LLVM::ConstantOp>( in matchAndRewrite() local
291 enable_ftz, index_64bit, cpu_codegen})); in matchAndRewrite()
|