Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A 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_;
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcompiler_functor.h44 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_;
H A 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,
H A Dsimple_orc_jit.cc106 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()
H A Dcpu_compiler.cc739 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()
/aosp_15_r20/external/pytorch/torch/csrc/jit/testing/
H A Dhooks_for_testing.h11 using ModuleHook = std::function<void(Module module)>; variable
16 TORCH_API void setEmitHooks(ModuleHook for_module, FunctionHook for_fn);
18 TORCH_API std::pair<ModuleHook, FunctionHook> getEmitHooks();
H A Dhooks_for_testing.cpp8 static ModuleHook emit_module_callback;
22 void setEmitHooks(ModuleHook for_mod, FunctionHook for_fn) { in setEmitHooks()
27 std::pair<ModuleHook, FunctionHook> getEmitHooks() { in getEmitHooks()
/aosp_15_r20/external/pytorch/torch/_C/
H A D__init__.pyi.in449 ModuleHook: Optional[Callable],