Crate caffe2op_gru

source ·

Structs


  • | GRUUnit computes the activations of | a standard GRU, in a sequence-length | aware fashion. | | Concretely, given the (fused) inputs | X (TxNxD), the previous hidden state | (NxD), and the sequence lengths (N), | computes the GRU activations, avoiding | computation if the input is invalid | (as in, the value at X[t][n] >= seqLengths[n]. |

Functions