#pragma once #include namespace torch::jit { TORCH_API void CanonicalizeOps(const std::shared_ptr& graph); }