Struct tensorflow::ImportGraphDefOptions

pub struct ImportGraphDefOptions { /* fields omitted */ }

ImportGraphDefOptions holds options that can be passed to Graph::import_graph_def.

Methods

impl ImportGraphDefOptions

pub fn new() -> Self

Creates a default ImportGraphDefOptions.

impl ImportGraphDefOptions

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.

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.

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.

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.

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.

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().

pub fn num_return_outputs(&self) -> usize

Returns the number of return outputs added via add_return_output().

pub fn num_return_operations(&self) -> usize

Returns the number of return operations added via add_return_operation().

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.

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.

pub fn set_default_device(&mut self, device: &str) -> Result<(), NulError>

Set the execution device for nodes. Only applies to nodes where a device was not already explicitly specified.

Trait Implementations

impl Drop for ImportGraphDefOptions

fn drop(&mut self)

Executes the destructor for this type.

impl Debug for ImportGraphDefOptions

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

Formats the value using the given formatter.