#pragma once #include #include namespace torch::autograd { using InferenceMode = c10::InferenceMode; }