Searched refs:kHiddenStateInTensor (Results 1 – 2 of 2) sorted by relevance
49 static constexpr int kHiddenStateInTensor = 4; variable
35 hidden_state_in_ = GetInput(operation, operands, kHiddenStateInTensor); in RNN()