pub struct TorchScriptImporter { /* private fields */ }Expand description
TorchScript importer
Implementations§
Source§impl TorchScriptImporter
impl TorchScriptImporter
pub fn new() -> Self
Sourcepub fn import_model(&self, model: &TorchScriptModel) -> Result<FxGraph>
pub fn import_model(&self, model: &TorchScriptModel) -> Result<FxGraph>
Import a TorchScript model into an FX graph
Sourcepub fn import_graph(&self, ts_graph: &TorchScriptGraph) -> Result<FxGraph>
pub fn import_graph(&self, ts_graph: &TorchScriptGraph) -> Result<FxGraph>
Import a TorchScript graph into an FX graph
Sourcepub fn add_operator_mapping(&mut self, torchscript_op: String, fx_op: String)
pub fn add_operator_mapping(&mut self, torchscript_op: String, fx_op: String)
Add custom operator mapping
Trait Implementations§
Auto Trait Implementations§
impl Freeze for TorchScriptImporter
impl RefUnwindSafe for TorchScriptImporter
impl Send for TorchScriptImporter
impl Sync for TorchScriptImporter
impl Unpin for TorchScriptImporter
impl UnsafeUnpin for TorchScriptImporter
impl UnwindSafe for TorchScriptImporter
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more