RNNCcell

associatedtype TimeStepInput : Differentiable where Self.Input == RNNCellInput<Self.TimeStepInput, Self.State>

associatedtype TimeStepOutput : Differentiable where Self.Output == RNNCellOutput<Self.TimeStepOutput, Self.State>

associatedtype State : Differentiable

func zeroState(for input: TimeStepInput) -> State

@differentiable
public func callAsFunction(
  input: TimeStepInput,
  state: State
) -> RNNCellOutput<TimeStepOutput, State>

@differentiable
public func call(input: TimeStepInput, state: State) -> RNNCellOutput<TimeStepOutput, State>