Struct Session

pub struct Session { /* private fields */ }

Manages a single graph and execution.

Implementations

impl Session

pub fn new(options: &SessionOptions, graph: &Graph) -> Result<Self>

Creates a session. graph will be be kept alive for the lifetime of the returned session. New nodes can still be added to graph after this call.

pub fn from_saved_model<P: AsRef<Path>, Tag: AsRef<str>, Tags: IntoIterator<Item = Tag>>( options: &SessionOptions, tags: Tags, graph: &mut Graph, export_dir: P, ) -> Result<Self>

👎Deprecated since 0.17.0: Please use SavedModelBundle::load() instead

Loads a session from an exported model.

pub fn close(&mut self) -> Result<()>

Closes the session.

pub fn run(&self, step: &mut SessionRunArgs<'_>) -> Result<()>

Runs the graph, feeding the inputs and then fetching the outputs requested in the step. Note that the session has interior mutability; this may mutate variables in the graph, and the caller is responsible for handling race conditions.

pub fn device_list(&self) -> Result<Vec<Device>>

Lists all devices in a session.

Trait Implementations

impl Debug for Session

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Drop for Session

fn drop(&mut self)

Executes the destructor for this type.

impl Send for Session

impl Sync for Session