Lines Matching refs:GetInput
55 input_ = GetInput(operation, operands, kInputTensor); in LSTMCell()
58 GetInput(operation, operands, kInputToInputWeightsTensor); // optional in LSTMCell()
59 input_to_forget_weights_ = GetInput(operation, operands, kInputToForgetWeightsTensor); in LSTMCell()
60 input_to_cell_weights_ = GetInput(operation, operands, kInputToCellWeightsTensor); in LSTMCell()
61 input_to_output_weights_ = GetInput(operation, operands, kInputToOutputWeightsTensor); in LSTMCell()
64 GetInput(operation, operands, kRecurrentToInputWeightsTensor); // optional in LSTMCell()
65 recurrent_to_forget_weights_ = GetInput(operation, operands, kRecurrentToForgetWeightsTensor); in LSTMCell()
66 recurrent_to_cell_weights_ = GetInput(operation, operands, kRecurrentToCellWeightsTensor); in LSTMCell()
67 recurrent_to_output_weights_ = GetInput(operation, operands, kRecurrentToOutputWeightsTensor); in LSTMCell()
69 cell_to_input_weights_ = GetInput(operation, operands, kCellToInputWeightsTensor); // optional in LSTMCell()
71 GetInput(operation, operands, kCellToForgetWeightsTensor); // optional in LSTMCell()
73 GetInput(operation, operands, kCellToOutputWeightsTensor); // optional in LSTMCell()
75 input_gate_bias_ = GetInput(operation, operands, kInputGateBiasTensor); in LSTMCell()
76 forget_gate_bias_ = GetInput(operation, operands, kForgetGateBiasTensor); in LSTMCell()
77 cell_bias_ = GetInput(operation, operands, kCellGateBiasTensor); in LSTMCell()
78 output_gate_bias_ = GetInput(operation, operands, kOutputGateBiasTensor); in LSTMCell()
80 projection_weights_ = GetInput(operation, operands, kProjectionWeightsTensor); // optional in LSTMCell()
81 projection_bias_ = GetInput(operation, operands, kProjectionBiasTensor); // optional in LSTMCell()
83 output_state_in_ = GetInput(operation, operands, kOutputStateInTensor); in LSTMCell()
84 cell_state_in_ = GetInput(operation, operands, kCellStateInTensor); in LSTMCell()
86 const auto& activationOperand = *GetInput(operation, operands, kActivationParam); in LSTMCell()
90 const auto& cellClipOperand = *GetInput(operation, operands, kCellClipParam); in LSTMCell()
91 const auto& projClipOperand = *GetInput(operation, operands, kProjClipParam); in LSTMCell()
106 GetInput(operation, operands, kInputLayerNormWeightsTensor); // optional in LSTMCell()
108 GetInput(operation, operands, kForgetLayerNormWeightsTensor); // optional in LSTMCell()
110 GetInput(operation, operands, kCellLayerNormWeightsTensor); // optional in LSTMCell()
112 GetInput(operation, operands, kOutputLayerNormWeightsTensor); // optional in LSTMCell()
327 NN_RET_CHECK(!IsNullInput(GetInput(operation, operands, requiredInput))) in Prepare()
333 const auto& activationOperand = *GetInput(operation, operands, kActivationParam); in Prepare()
335 const auto& cellClipOperand = *GetInput(operation, operands, kCellClipParam); in Prepare()
336 const auto& projClipOperand = *GetInput(operation, operands, kProjClipParam); in Prepare()