Struct ssvm_tensorflow_interface::Session [−][src]
pub struct Session { /* fields omitted */ }
Expand description
The session structure.
Implementations
impl Session
[src]
impl Session
[src]pub fn new<S: AsRef<[u8]>>(model_buf: S, mod_type: ModelType) -> Session
[src]
pub fn add_input<T: TensorType>(
&mut self,
name: &str,
tensor_buf: &[T],
shape: &[i64]
) -> &mut Session
[src]
pub fn add_input<T: TensorType>(
&mut self,
name: &str,
tensor_buf: &[T],
shape: &[i64]
) -> &mut Session
[src]Add input name, dimension, operation index, and input tensor into context.
pub fn add_output(&mut self, name: &str) -> &mut Session
[src]
pub fn add_output(&mut self, name: &str) -> &mut Session
[src]Add output name and operation index into context.
pub fn clear_input(&mut self) -> &mut Session
[src]
pub fn clear_input(&mut self) -> &mut Session
[src]Clear the set input tensors.
pub fn clear_output(&mut self) -> &mut Session
[src]
pub fn clear_output(&mut self) -> &mut Session
[src]Clear the set output tensors.
pub fn get_output<T: TensorType>(&self, name: &str) -> Vec<T>
[src]
pub fn get_output<T: TensorType>(&self, name: &str) -> Vec<T>
[src]Get output tensor data by name.