Searched refs:kInputToCellWeightsTensor (Results 1 – 2 of 2) sorted by relevance
67 static constexpr int kInputToCellWeightsTensor = 3; variable
45 input_to_cell_weights_ = GetInput(operation, operands, kInputToCellWeightsTensor); in LSTMCell()112 GetInput(operation, operands, LSTMCell::kInputToCellWeightsTensor); in CheckInputTensorDimensions()