Searched refs:kRecurrentToOutputWeightsTensor (Results 1 – 2 of 2) sorted by relevance
74 static constexpr int kRecurrentToOutputWeightsTensor = 8; variable
52 recurrent_to_output_weights_ = GetInput(operation, operands, kRecurrentToOutputWeightsTensor); in LSTMCell()256 GetInput(operation, operands, LSTMCell::kRecurrentToOutputWeightsTensor); in Prepare()