Struct tensorflow::WhileBuilder

pub struct WhileBuilder<'a> { /* private fields */ }

A WhileBuilder is used to build a while loop.

Implementations

impl<'a> WhileBuilder<'a>

pub fn new<CF: Fn(&mut Graph, &[Output]) -> Result<Output>, BF: Fn(&mut Graph, &[Output]) -> Result<Vec<Output>>>( graph: &'a mut Graph, cond: CF, body: BF, inputs: &[Output] ) -> Result<Self>

Creates a WhileBuilder for creating a while loop.

The loop is not actually added to the graph until finish is called.

Arguments

• graph - graph to create the while loop in
• cond - takes the condition subgraph and loop variables and returns a scalar boolean
• body - takes the body subgraph and loop variables and returns the new values of the loop variables
• inputs - outputs that already exist in graph used as initial values for the loop variables

Returns

A unfinished WhileBuilder.

pub fn name(self, name: &str) -> Result<Self, NulError>

Sets a unique name for this while loop. This is used as a prefix for created operations. If not set, a unique prefix will be generated.

pub fn finish(self) -> Result<Vec<Output>>

Builds the while loop and returns the output tensors of the while loop.

Trait Implementations

impl<'a> Debug for WhileBuilder<'a>

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

Formats the value using the given formatter.