Lines Matching refs:execution

93         Execution execution(&compilation);  in invoke()  local
96 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kInputTensor, input_), in invoke()
98 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kInputToInputWeightsTensor, in invoke()
101 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kInputToForgetWeightsTensor, in invoke()
104 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kInputToCellWeightsTensor, in invoke()
107 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kInputToOutputWeightsTensor, in invoke()
110 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kRecurrentToInputWeightsTensor, in invoke()
113 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kRecurrentToForgetWeightsTensor, in invoke()
116 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kRecurrentToCellWeightsTensor, in invoke()
119 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kRecurrentToOutputWeightsTensor, in invoke()
123 setInputTensor(&execution, QuantizedLSTMCell::kInputGateBiasTensor, inputGateBias_), in invoke()
125 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kForgetGateBiasTensor, in invoke()
128 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kCellGateBiasTensor, cellGateBias_), in invoke()
130 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kOutputGateBiasTensor, in invoke()
134 setInputTensor(&execution, QuantizedLSTMCell::kPrevCellStateTensor, prevCellState_), in invoke()
136 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kPrevOutputTensor, prevOutput_), in invoke()
140 setOutputTensor(&execution, QuantizedLSTMCell::kCellStateOutTensor, &cellStateOut_), in invoke()
142 ASSERT_EQ(setOutputTensor(&execution, QuantizedLSTMCell::kOutputTensor, &output_), in invoke()
145 ASSERT_EQ(execution.compute(), Result::NO_ERROR); in invoke()
225 Result setInputTensor(Execution* execution, int tensor, const std::vector<T>& data) { in setInputTensor() argument
226 return execution->setInput(tensor, data.data(), sizeof(T) * data.size()); in setInputTensor()
229 Result setOutputTensor(Execution* execution, int tensor, std::vector<T>* data) { in setOutputTensor() argument
230 return execution->setOutput(tensor, data->data(), sizeof(T) * data->size()); in setOutputTensor()