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()
88 ModuleHook user_pre_optimization_hook_;
89 ModuleHook user_post_optimization_hook_;
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcompiler_functor.h37 LLVMCompiler::ModuleHook pre_optimization_hook = nullptr,
38 LLVMCompiler::ModuleHook post_optimization_hook = nullptr,
69 LLVMCompiler::ModuleHook pre_optimization_hook_;
70 LLVMCompiler::ModuleHook post_optimization_hook_;
Dsimple_orc_jit.h61 LLVMCompiler::ModuleHook pre_optimization_hook,
62 LLVMCompiler::ModuleHook post_optimization_hook,
Dcpu_compiler.cc444 std::pair<LLVMCompiler::ModuleHook, LLVMCompiler::ModuleHook> GetIRModuleHooks( in GetIRModuleHooks()
446 const LLVMCompiler::ModuleHook& user_pre_optimization_hook, in GetIRModuleHooks()
447 const LLVMCompiler::ModuleHook& user_post_optimization_hook) { in GetIRModuleHooks()
595 ModuleHook pre_optimization_ir_hook; in RunBackend()
596 ModuleHook post_optimization_ir_hook; in RunBackend()
887 ModuleHook pre_optimization_ir_hook; in CompileAheadOfTime()
888 ModuleHook post_optimization_ir_hook; in CompileAheadOfTime()
Dsimple_orc_jit.cc98 LLVMCompiler::ModuleHook pre_optimization_hook, in SimpleOrcJIT()
99 LLVMCompiler::ModuleHook post_optimization_hook, in SimpleOrcJIT()