#pragma once #include #include #include #include namespace torch { namespace jit { using PrintHandler = void (*)(const std::string&); TORCH_API void setPrintHandler(PrintHandler ph); TORCH_API PrintHandler getPrintHandler(); } // namespace jit } // namespace torch