Home
last modified time | relevance | path

Searched refs:ModuleHook (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dllvm_compiler.h40 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/
Dcompiler_functor.h39 LLVMCompiler::ModuleHook pre_optimization_hook = nullptr,
40 LLVMCompiler::ModuleHook post_optimization_hook = nullptr,
72 LLVMCompiler::ModuleHook pre_optimization_hook_;
73 LLVMCompiler::ModuleHook post_optimization_hook_;
Dsimple_orc_jit.h62 LLVMCompiler::ModuleHook pre_optimization_hook,
63 LLVMCompiler::ModuleHook post_optimization_hook,
70 LLVMCompiler::ModuleHook pre_optimization_hook,
71 LLVMCompiler::ModuleHook post_optimization_hook,
Dsimple_orc_jit.cc93 LLVMCompiler::ModuleHook pre_optimization_hook, in SimpleOrcJIT()
94 LLVMCompiler::ModuleHook post_optimization_hook, in SimpleOrcJIT()
163 LLVMCompiler::ModuleHook pre_optimization_hook, in Create()
164 LLVMCompiler::ModuleHook post_optimization_hook, in Create()
Dcpu_compiler.cc494 std::pair<LLVMCompiler::ModuleHook, LLVMCompiler::ModuleHook> GetIRModuleHooks( in GetIRModuleHooks()
496 const LLVMCompiler::ModuleHook& user_pre_optimization_hook, in GetIRModuleHooks()
497 const LLVMCompiler::ModuleHook& user_post_optimization_hook) { in GetIRModuleHooks()
668 ModuleHook pre_optimization_ir_hook; in RunBackend()
669 ModuleHook post_optimization_ir_hook; in RunBackend()
977 ModuleHook pre_optimization_ir_hook; in CompileAheadOfTime()
978 ModuleHook post_optimization_ir_hook; in CompileAheadOfTime()