Struct DefaultSession 

pub struct DefaultSession { /* private fields */ }

Default session implementation

Implementations

impl DefaultSession

pub fn new( graph: Arc<RwLock<Graph>>, config: SessionConfig, op_registry: Arc<OpRegistry>, ) -> Self

Create a new session with the given graph and configuration

Trait Implementations

impl Session for DefaultSession

fn run( &mut self, fetches: &[FetchSpec], feed_dict: &FeedDict, ) -> Result<Vec<Tensor<f32>>, TensorError>

Run the graph with the given feeds and fetches

fn partial_run_setup( &mut self, feeds: &[String], fetches: &[FetchSpec], targets: &[String], ) -> Result<String, TensorError>

Partial run - allows incrementally feeding inputs and fetching outputs

fn partial_run( &mut self, handle: &str, feed_dict: &FeedDict, fetches: &[FetchSpec], ) -> Result<Vec<Tensor<f32>>, TensorError>

Continue a partial run

fn close(&mut self) -> Result<(), TensorError>

Close the session and release resources