Struct LSTMBlockCell

pub struct LSTMBlockCell { /* private fields */ }

Builder for the LSTMBlockCell operation.

Implementations

impl LSTMBlockCell

pub fn new() -> Self

Creates a new LSTMBlockCell.

pub fn forget_bias<ArgType: Into<f32>>(self, value: ArgType) -> Self

Sets the forget_bias attribute.

pub fn cell_clip<ArgType: Into<f32>>(self, value: ArgType) -> Self

Sets the cell_clip attribute.

pub fn use_peephole<ArgType: Into<bool>>(self, value: ArgType) -> Self

Sets the use_peephole attribute.

pub fn T<ArgType: Into<DataType>>(self, value: ArgType) -> Self

Sets the T attribute.

pub fn add_control_input(self, op: Operation) -> Self

Adds a control input.

pub fn build<O0: Into<Output>, O1: Into<Output>, O2: Into<Output>, O3: Into<Output>, O4: Into<Output>, O5: Into<Output>, O6: Into<Output>, O7: Into<Output>>( &self, x: O0, cs_prev: O1, h_prev: O2, w: O3, wci: O4, wcf: O5, wco: O6, b: O7, scope: &mut Scope, ) -> Result<Operation>

Builds the LSTMBlockCell operation.

Trait Implementations

impl Debug for LSTMBlockCell

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

Formats the value using the given formatter.

impl Default for LSTMBlockCell

fn default() -> LSTMBlockCell

Returns the “default value” for a type.