• Home
  • Raw
  • Download

Lines Matching refs:GetInput

60     input_ = GetInput(operation, operands, kInputTensor);  in LSTMCell()
63 GetInput(operation, operands, kInputToInputWeightsTensor); // optional in LSTMCell()
64 input_to_forget_weights_ = GetInput(operation, operands, kInputToForgetWeightsTensor); in LSTMCell()
65 input_to_cell_weights_ = GetInput(operation, operands, kInputToCellWeightsTensor); in LSTMCell()
66 input_to_output_weights_ = GetInput(operation, operands, kInputToOutputWeightsTensor); in LSTMCell()
69 GetInput(operation, operands, kRecurrentToInputWeightsTensor); // optional in LSTMCell()
70 recurrent_to_forget_weights_ = GetInput(operation, operands, kRecurrentToForgetWeightsTensor); in LSTMCell()
71 recurrent_to_cell_weights_ = GetInput(operation, operands, kRecurrentToCellWeightsTensor); in LSTMCell()
72 recurrent_to_output_weights_ = GetInput(operation, operands, kRecurrentToOutputWeightsTensor); in LSTMCell()
74 cell_to_input_weights_ = GetInput(operation, operands, kCellToInputWeightsTensor); // optional in LSTMCell()
76 GetInput(operation, operands, kCellToForgetWeightsTensor); // optional in LSTMCell()
78 GetInput(operation, operands, kCellToOutputWeightsTensor); // optional in LSTMCell()
80 input_gate_bias_ = GetInput(operation, operands, kInputGateBiasTensor); in LSTMCell()
81 forget_gate_bias_ = GetInput(operation, operands, kForgetGateBiasTensor); in LSTMCell()
82 cell_bias_ = GetInput(operation, operands, kCellGateBiasTensor); in LSTMCell()
83 output_gate_bias_ = GetInput(operation, operands, kOutputGateBiasTensor); in LSTMCell()
85 projection_weights_ = GetInput(operation, operands, kProjectionWeightsTensor); // optional in LSTMCell()
86 projection_bias_ = GetInput(operation, operands, kProjectionBiasTensor); // optional in LSTMCell()
88 output_state_in_ = GetInput(operation, operands, kOutputStateInTensor); in LSTMCell()
89 cell_state_in_ = GetInput(operation, operands, kCellStateInTensor); in LSTMCell()
91 const auto& activationOperand = *GetInput(operation, operands, kActivationParam); in LSTMCell()
95 const auto& cellClipOperand = *GetInput(operation, operands, kCellClipParam); in LSTMCell()
96 const auto& projClipOperand = *GetInput(operation, operands, kProjClipParam); in LSTMCell()
111 GetInput(operation, operands, kInputLayerNormWeightsTensor); // optional in LSTMCell()
113 GetInput(operation, operands, kForgetLayerNormWeightsTensor); // optional in LSTMCell()
115 GetInput(operation, operands, kCellLayerNormWeightsTensor); // optional in LSTMCell()
117 GetInput(operation, operands, kOutputLayerNormWeightsTensor); // optional in LSTMCell()
332 NN_RET_CHECK(!IsNullInput(GetInput(operation, operands, requiredInput))) in Prepare()
338 const auto& activationOperand = *GetInput(operation, operands, kActivationParam); in Prepare()
340 const auto& cellClipOperand = *GetInput(operation, operands, kCellClipParam); in Prepare()
341 const auto& projClipOperand = *GetInput(operation, operands, kProjClipParam); in Prepare()