Lines Matching refs:GetInput

82     input_ = GetInput(operation, operands, kInputTensor);  in BidirectionalSequenceLSTM()
85 GetInput(operation, operands, kFwInputToInputWeightsTensor); // optional in BidirectionalSequenceLSTM()
86 fw_input_to_forget_weights_ = GetInput(operation, operands, kFwInputToForgetWeightsTensor); in BidirectionalSequenceLSTM()
87 fw_input_to_cell_weights_ = GetInput(operation, operands, kFwInputToCellWeightsTensor); in BidirectionalSequenceLSTM()
88 fw_input_to_output_weights_ = GetInput(operation, operands, kFwInputToOutputWeightsTensor); in BidirectionalSequenceLSTM()
91 GetInput(operation, operands, kFwRecurrentToInputWeightsTensor); // optional in BidirectionalSequenceLSTM()
93 GetInput(operation, operands, kFwRecurrentToForgetWeightsTensor); in BidirectionalSequenceLSTM()
94 fw_recurrent_to_cell_weights_ = GetInput(operation, operands, kFwRecurrentToCellWeightsTensor); in BidirectionalSequenceLSTM()
96 GetInput(operation, operands, kFwRecurrentToOutputWeightsTensor); in BidirectionalSequenceLSTM()
99 GetInput(operation, operands, kFwCellToInputWeightsTensor); // optional in BidirectionalSequenceLSTM()
101 GetInput(operation, operands, kFwCellToForgetWeightsTensor); // optional in BidirectionalSequenceLSTM()
103 GetInput(operation, operands, kFwCellToOutputWeightsTensor); // optional in BidirectionalSequenceLSTM()
105 fw_input_gate_bias_ = GetInput(operation, operands, kFwInputGateBiasTensor); in BidirectionalSequenceLSTM()
106 fw_forget_gate_bias_ = GetInput(operation, operands, kFwForgetGateBiasTensor); in BidirectionalSequenceLSTM()
107 fw_cell_bias_ = GetInput(operation, operands, kFwCellGateBiasTensor); in BidirectionalSequenceLSTM()
108 fw_output_gate_bias_ = GetInput(operation, operands, kFwOutputGateBiasTensor); in BidirectionalSequenceLSTM()
110 fw_projection_weights_ = GetInput(operation, operands, kFwProjectionWeightsTensor); // optional in BidirectionalSequenceLSTM()
111 fw_projection_bias_ = GetInput(operation, operands, kFwProjectionBiasTensor); // optional in BidirectionalSequenceLSTM()
113 fw_activation_state_ = GetInput(operation, operands, kFwInputActivationStateTensor); in BidirectionalSequenceLSTM()
114 fw_cell_state_ = GetInput(operation, operands, kFwInputCellStateTensor); in BidirectionalSequenceLSTM()
117 GetInput(operation, operands, kBwInputToInputWeightsTensor); // optional in BidirectionalSequenceLSTM()
118 bw_input_to_forget_weights_ = GetInput(operation, operands, kBwInputToForgetWeightsTensor); in BidirectionalSequenceLSTM()
119 bw_input_to_cell_weights_ = GetInput(operation, operands, kBwInputToCellWeightsTensor); in BidirectionalSequenceLSTM()
120 bw_input_to_output_weights_ = GetInput(operation, operands, kBwInputToOutputWeightsTensor); in BidirectionalSequenceLSTM()
123 GetInput(operation, operands, kBwRecurrentToInputWeightsTensor); // optional in BidirectionalSequenceLSTM()
125 GetInput(operation, operands, kBwRecurrentToForgetWeightsTensor); in BidirectionalSequenceLSTM()
126 bw_recurrent_to_cell_weights_ = GetInput(operation, operands, kBwRecurrentToCellWeightsTensor); in BidirectionalSequenceLSTM()
128 GetInput(operation, operands, kBwRecurrentToOutputWeightsTensor); in BidirectionalSequenceLSTM()
131 GetInput(operation, operands, kBwCellToInputWeightsTensor); // optional in BidirectionalSequenceLSTM()
133 GetInput(operation, operands, kBwCellToForgetWeightsTensor); // optional in BidirectionalSequenceLSTM()
135 GetInput(operation, operands, kBwCellToOutputWeightsTensor); // optional in BidirectionalSequenceLSTM()
137 bw_input_gate_bias_ = GetInput(operation, operands, kBwInputGateBiasTensor); in BidirectionalSequenceLSTM()
138 bw_forget_gate_bias_ = GetInput(operation, operands, kBwForgetGateBiasTensor); in BidirectionalSequenceLSTM()
139 bw_cell_bias_ = GetInput(operation, operands, kBwCellGateBiasTensor); in BidirectionalSequenceLSTM()
140 bw_output_gate_bias_ = GetInput(operation, operands, kBwOutputGateBiasTensor); in BidirectionalSequenceLSTM()
142 bw_projection_weights_ = GetInput(operation, operands, kBwProjectionWeightsTensor); // optional in BidirectionalSequenceLSTM()
143 bw_projection_bias_ = GetInput(operation, operands, kBwProjectionBiasTensor); // optional in BidirectionalSequenceLSTM()
145 bw_activation_state_ = GetInput(operation, operands, kBwInputActivationStateTensor); in BidirectionalSequenceLSTM()
146 bw_cell_state_ = GetInput(operation, operands, kBwInputCellStateTensor); in BidirectionalSequenceLSTM()
148 aux_input_ = GetInput(operation, operands, kAuxInputTensor); in BidirectionalSequenceLSTM()
149 fw_aux_input_to_input_weights_ = GetInput(operation, operands, kFwAuxInputToInputWeightsTensor); in BidirectionalSequenceLSTM()
151 GetInput(operation, operands, kFwAuxInputToForgetWeightsTensor); in BidirectionalSequenceLSTM()
152 fw_aux_input_to_cell_weights_ = GetInput(operation, operands, kFwAuxInputToCellWeightsTensor); in BidirectionalSequenceLSTM()
154 GetInput(operation, operands, kFwAuxInputToOutputWeightsTensor); in BidirectionalSequenceLSTM()
155 bw_aux_input_to_input_weights_ = GetInput(operation, operands, kBwAuxInputToInputWeightsTensor); in BidirectionalSequenceLSTM()
157 GetInput(operation, operands, kBwAuxInputToForgetWeightsTensor); in BidirectionalSequenceLSTM()
158 bw_aux_input_to_cell_weights_ = GetInput(operation, operands, kBwAuxInputToCellWeightsTensor); in BidirectionalSequenceLSTM()
160 GetInput(operation, operands, kBwAuxInputToOutputWeightsTensor); in BidirectionalSequenceLSTM()
162 fw_input_layer_norm_weights_ = GetInput(operation, operands, kFwInputLayerNormWeightsTensor); in BidirectionalSequenceLSTM()
163 fw_forget_layer_norm_weights_ = GetInput(operation, operands, kFwForgetLayerNormWeightsTensor); in BidirectionalSequenceLSTM()
164 fw_cell_layer_norm_weights_ = GetInput(operation, operands, kFwCellLayerNormWeightsTensor); in BidirectionalSequenceLSTM()
165 fw_output_layer_norm_weights_ = GetInput(operation, operands, kFwOutputLayerNormWeightsTensor); in BidirectionalSequenceLSTM()
166 bw_input_layer_norm_weights_ = GetInput(operation, operands, kBwInputLayerNormWeightsTensor); in BidirectionalSequenceLSTM()
167 bw_forget_layer_norm_weights_ = GetInput(operation, operands, kBwForgetLayerNormWeightsTensor); in BidirectionalSequenceLSTM()
168 bw_cell_layer_norm_weights_ = GetInput(operation, operands, kBwCellLayerNormWeightsTensor); in BidirectionalSequenceLSTM()
169 bw_output_layer_norm_weights_ = GetInput(operation, operands, kBwOutputLayerNormWeightsTensor); in BidirectionalSequenceLSTM()
171 const auto& activationOperand = *GetInput(operation, operands, kActivationParam); in BidirectionalSequenceLSTM()
174 const auto& clipOperand = *GetInput(operation, operands, kCellClipParam); in BidirectionalSequenceLSTM()
175 const auto& projOperand = *GetInput(operation, operands, kProjClipParam); in BidirectionalSequenceLSTM()
185 const auto& mergeOutputsOperand = *GetInput(operation, operands, kMergeOutputsParam); in BidirectionalSequenceLSTM()
187 const auto& timeMajorOperand = *GetInput(operation, operands, kTimeMajorParam); in BidirectionalSequenceLSTM()
244 NN_RET_CHECK(!IsNullInput(GetInput(operation, operands, requiredInput))) in Prepare()
249 const auto& activationOperand = *GetInput(operation, operands, kActivationParam); in Prepare()
251 const auto& cellOperand = *GetInput(operation, operands, kCellClipParam); in Prepare()
252 const auto& projOperand = *GetInput(operation, operands, kProjClipParam); in Prepare()
260 const auto& mergeOutputsOperand = *GetInput(operation, operands, kMergeOutputsParam); in Prepare()
262 const auto& timeMajorOperand = *GetInput(operation, operands, kTimeMajorParam); in Prepare()