/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/testing
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/testing/hooks_for_testing.h (617B)
#pragma once
#include
#include
#include
#include
namespace torch {
namespace jit {
struct Module;
using ModuleHook = std::function;
using FunctionHook = std::function;
TORCH_API void didFinishEmitModule(Module module);
TORCH_API void didFinishEmitFunction(StrongFunctionPtr defined);
TORCH_API void setEmitHooks(ModuleHook for_module, FunctionHook for_fn);
TORCH_API std::pair getEmitHooks();
} // namespace jit
} // namespace torch