torch/csrc/mps/Module.h

#pragma once

#include <torch/csrc/python_headers.h>

namespace torch {
namespace mps {

PyMethodDef* python_functions();

} // namespace mps
} // namespace torch