mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
c8d1ec02be
* Have ScriptModule inherit from Module This is accomplished by created replacement _parameters, _buffers, and _modules which implement the OrderedDict APIs but which actually get/set their members inside script::Module * Merge TracedModule with ScriptModule * Move logic of attribute handling into Python bindings rather than make script::Module handle it. This was redundant with nn.Module, which already handles attribute. * Make TracedModule a subclass of ScriptModule * Move handling of attribute kind logic into bindings. * Allow ScriptModule to contain non-script module submodules.
13 lines
212 B
C++
13 lines
212 B
C++
#pragma once
|
|
|
|
#include "torch/csrc/jit/pybind.h"
|
|
|
|
namespace torch {
|
|
namespace jit {
|
|
namespace script {
|
|
void initJitScriptBindings(PyObject* module);
|
|
|
|
} // namespace script
|
|
} // namespace jit
|
|
} // namespace torch
|