Lines Matching refs:vector
33 std::vector<uint32_t> shape;
37 OperandTypeParams(Type type, std::vector<uint32_t> shape, float scale, int32_t zeroPoint) in OperandTypeParams()
50 QuantizedLSTMOpModel(const std::vector<OperandTypeParams>& inputOperandTypeParams) { in QuantizedLSTMOpModel()
51 std::vector<uint32_t> inputs; in QuantizedLSTMOpModel()
65 std::vector<uint32_t> outputs; in QuantizedLSTMOpModel()
156 void setInput(const std::vector<uint8_t>& input) { input_ = input; } in setInput()
158 void setWeightsAndBiases(std::vector<uint8_t> inputToInputWeights, in setWeightsAndBiases()
159 std::vector<uint8_t> inputToForgetWeights, in setWeightsAndBiases()
160 std::vector<uint8_t> inputToCellWeights, in setWeightsAndBiases()
161 std::vector<uint8_t> inputToOutputWeights, in setWeightsAndBiases()
162 std::vector<uint8_t> recurrentToInputWeights, in setWeightsAndBiases()
163 std::vector<uint8_t> recurrentToForgetWeights, in setWeightsAndBiases()
164 std::vector<uint8_t> recurrentToCellWeights, in setWeightsAndBiases()
165 std::vector<uint8_t> recurrentToOutputWeights, in setWeightsAndBiases()
166 std::vector<int32_t> inputGateBias, in setWeightsAndBiases()
167 std::vector<int32_t> forgetGateBias, in setWeightsAndBiases()
168 std::vector<int32_t> cellGateBias, // in setWeightsAndBiases()
169 std::vector<int32_t> outputGateBias) { in setWeightsAndBiases()
185 void initializeInputData(OperandTypeParams params, std::vector<T>* vec) { in initializeInputData()
194 std::vector<uint8_t> getOutput() { return output_; } in getOutput()
202 std::vector<uint8_t> input_;
203 std::vector<uint8_t> inputToInputWeights_;
204 std::vector<uint8_t> inputToForgetWeights_;
205 std::vector<uint8_t> inputToCellWeights_;
206 std::vector<uint8_t> inputToOutputWeights_;
207 std::vector<uint8_t> recurrentToInputWeights_;
208 std::vector<uint8_t> recurrentToForgetWeights_;
209 std::vector<uint8_t> recurrentToCellWeights_;
210 std::vector<uint8_t> recurrentToOutputWeights_;
211 std::vector<int32_t> inputGateBias_;
212 std::vector<int32_t> forgetGateBias_;
213 std::vector<int32_t> cellGateBias_;
214 std::vector<int32_t> outputGateBias_;
215 std::vector<int16_t> prevCellState_;
216 std::vector<uint8_t> prevOutput_;
218 std::vector<int16_t> cellStateOut_;
219 std::vector<uint8_t> output_;
225 Result setInputTensor(Execution* execution, int tensor, const std::vector<T>& data) { in setInputTensor()
229 Result setOutputTensor(Execution* execution, int tensor, std::vector<T>* data) { in setOutputTensor()
236 void VerifyGoldens(const std::vector<std::vector<uint8_t>>& input, in VerifyGoldens()
237 const std::vector<std::vector<uint8_t>>& output, in VerifyGoldens()
246 std::vector<uint8_t> inputStep; in VerifyGoldens()
256 std::vector<float> expected; in VerifyGoldens()
352 std::vector<std::vector<uint8_t>> lstmInput; in TEST_F()
363 std::vector<std::vector<uint8_t>> lstmGoldenOutput; in TEST_F()