mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
8460131087
Summary: Implement a ModelRunner from scratch with the minimum features for OSS only Test Plan: test_export -r NativeRT Rollback Plan: Differential Revision: D78979812 Pull Request resolved: https://github.com/pytorch/pytorch/pull/159268 Approved by: https://github.com/dolpm
14 lines
213 B
C++
14 lines
213 B
C++
#pragma once
|
|
|
|
#include <torch/csrc/utils/pybind.h>
|
|
|
|
namespace py = pybind11;
|
|
|
|
namespace torch {
|
|
namespace nativert {
|
|
|
|
void initModelRunnerPybind(pybind11::module& m);
|
|
|
|
} // namespace nativert
|
|
} // namespace torch
|