Struct tensorflow::ImportGraphDefOptions
source · pub struct ImportGraphDefOptions { /* private fields */ }Expand description
ImportGraphDefOptions holds options that can be passed to
Graph::import_graph_def.
Implementations§
source§impl ImportGraphDefOptions
impl ImportGraphDefOptions
sourcepub fn set_prefix(&mut self, prefix: &str) -> Result<(), NulError>
pub fn set_prefix(&mut self, prefix: &str) -> Result<(), NulError>
Set the prefix to be prepended to the names of nodes in graph_def that will
be imported into graph.
sourcepub fn add_input_mapping(
&mut self,
src_name: &str,
src_index: usize,
dst: &Output
) -> Result<(), NulError>
pub fn add_input_mapping( &mut self, src_name: &str, src_index: usize, dst: &Output ) -> Result<(), NulError>
Set any imported nodes with input src_name:src_index to have that input
replaced with dst. src_name refers to a node in the graph to be imported,
dst references a node already existing in the graph being imported into.
sourcepub fn remap_control_dependency(
&mut self,
src_name: &str,
dst: &Operation
) -> Result<(), NulError>
pub fn remap_control_dependency( &mut self, src_name: &str, dst: &Operation ) -> Result<(), NulError>
Set any imported nodes with control input src_name to have that input
replaced with dst. src_name refers to a node in the graph to be imported,
dst references an operation already existing in the graph being imported
into.
sourcepub fn add_control_dependency(&mut self, oper: &Operation)
pub fn add_control_dependency(&mut self, oper: &Operation)
Cause the imported graph to have a control dependency on oper. oper
should exist in the graph being imported into.
sourcepub fn add_return_output(
&mut self,
oper_name: &str,
index: usize
) -> Result<(), NulError>
pub fn add_return_output( &mut self, oper_name: &str, index: usize ) -> Result<(), NulError>
Add an output in graph_def to be returned via the return_outputs output
parameter of import_graph_def(). If the output is remapped via an input
mapping, the corresponding existing tensor in graph will be returned.
sourcepub fn add_return_operation(&mut self, oper_name: &str) -> Result<(), NulError>
pub fn add_return_operation(&mut self, oper_name: &str) -> Result<(), NulError>
Add an operation in graph_def to be returned via the return_opers output
parameter of import_graph_def().
sourcepub fn num_return_outputs(&self) -> usize
pub fn num_return_outputs(&self) -> usize
Returns the number of return outputs added via add_return_output().
sourcepub fn num_return_operations(&self) -> usize
pub fn num_return_operations(&self) -> usize
Returns the number of return operations added via add_return_operation().
sourcepub fn set_uniquify_names(&mut self, uniquify_names: bool)
pub fn set_uniquify_names(&mut self, uniquify_names: bool)
Set whether to uniquify imported operation names. If true, imported operation names will be modified if their name already exists in the graph. If false, conflicting names will be treated as an error. Note that this option has no effect if a prefix is set, since the prefix will guarantee all names are unique. Defaults to false.
sourcepub fn set_uniquify_prefix(&mut self, uniquify_prefix: bool)
pub fn set_uniquify_prefix(&mut self, uniquify_prefix: bool)
If true, the specified prefix will be modified if it already exists as an operation name or prefix in the graph. If false, a conflicting prefix will be treated as an error. This option has no effect if no prefix is specified.