Struct SignatureDef

pub struct SignatureDef { /* private fields */ }

SignatureDef defines the signature of a computation supported by a TensorFlow graph.

Implementations

impl SignatureDef

pub fn new(method_name: String) -> SignatureDef

Creates a SignatureDef with the given method name.

pub fn add_input_info(&mut self, name: String, info: TensorInfo)

Adds an input parameter.

pub fn add_output_info(&mut self, name: String, info: TensorInfo)

Adds an output parameter.

pub fn method_name(&self) -> &str

Returns the method name.

pub fn inputs(&self) -> &HashMap<String, TensorInfo>

Returns the input parameters.

pub fn outputs(&self) -> &HashMap<String, TensorInfo>

Returns the output parameters.

pub fn get_input(&self, name: &str) -> Result<&TensorInfo>

Returns the given input parameter.

pub fn get_output(&self, name: &str) -> Result<&TensorInfo>

Returns the given output parameter.

Trait Implementations

impl Clone for SignatureDef

fn clone(&self) -> SignatureDef

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for SignatureDef

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

Formats the value using the given formatter.