/
usr
/
local
/
lib64
/
python3.6
/
site-packages
/
torch
/
include
/
torch
/
csrc
/
jit
/
testing
/
/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/testing
mkdir
upload
Name
Size
Mode
Actions
file_check.h
2421
0644
edit
dl
rm
hooks_for_testing.h
617
0644
edit
dl
rm
Edit:
/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/testing/hooks_for_testing.h
(617B)
#pragma once #include <torch/csrc/WindowsTorchApiMacro.h> #include <torch/csrc/jit/api/compilation_unit.h> #include <functional> #include <memory> namespace torch { namespace jit { struct Module; using ModuleHook = std::function<void(Module module)>; using FunctionHook = std::function<void(StrongFunctionPtr 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<ModuleHook, FunctionHook> getEmitHooks(); } // namespace jit } // namespace torch
Save
cmd:
run