Searched refs:ModuleHook (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | llvm_compiler.h | 40 using ModuleHook = std::function<void(const llvm::Module&)>; variable
42 void SetPreOptimizationHook(ModuleHook hook) { in SetPreOptimizationHook()
51 void SetPostOptimizationHook(ModuleHook hook) { in SetPostOptimizationHook()
79 ModuleHook user_pre_optimization_hook_;
80 ModuleHook user_post_optimization_hook_;
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | compiler_functor.h | 44 LLVMCompiler::ModuleHook pre_optimization_hook = nullptr,
45 LLVMCompiler::ModuleHook post_optimization_hook = nullptr,
81 LLVMCompiler::ModuleHook pre_optimization_hook_;
82 LLVMCompiler::ModuleHook post_optimization_hook_;
|
| D | simple_orc_jit.h | 62 LLVMCompiler::ModuleHook pre_optimization_hook,
63 LLVMCompiler::ModuleHook post_optimization_hook,
70 LLVMCompiler::ModuleHook pre_optimization_hook,
71 LLVMCompiler::ModuleHook post_optimization_hook,
|
| D | simple_orc_jit.cc | 106 LLVMCompiler::ModuleHook pre_optimization_hook, in SimpleOrcJIT()
107 LLVMCompiler::ModuleHook post_optimization_hook, in SimpleOrcJIT()
177 LLVMCompiler::ModuleHook pre_optimization_hook, in Create()
178 LLVMCompiler::ModuleHook post_optimization_hook, in Create()
|
| D | cpu_compiler.cc | 739 std::pair<LLVMCompiler::ModuleHook, LLVMCompiler::ModuleHook> GetIRModuleHooks( in GetIRModuleHooks()
741 const LLVMCompiler::ModuleHook& user_pre_optimization_hook, in GetIRModuleHooks()
742 const LLVMCompiler::ModuleHook& user_post_optimization_hook) { in GetIRModuleHooks()
1118 ModuleHook pre_optimization_ir_hook; in CompileLegacyCpuExecutable()
1119 ModuleHook post_optimization_ir_hook; in CompileLegacyCpuExecutable()
1526 ModuleHook pre_optimization_ir_hook; in CompileAheadOfTime()
1527 ModuleHook post_optimization_ir_hook; in CompileAheadOfTime()
|