Lines Matching refs:execution
196 Execution execution(&compilation); in Invoke() local
199 execution.setInput(LSTMCell::k##X##Tensor, X##_.data(), sizeof(float) * X##_.size()), \ in Invoke()
208 execution.setOutput(LSTMCell::k##X##Tensor, X##_.data(), sizeof(float) * X##_.size()), \ in Invoke()
216 execution.setInput(LSTMCell::kInputToInputWeightsTensor, nullptr, 0); in Invoke()
217 execution.setInput(LSTMCell::kRecurrentToInputWeightsTensor, nullptr, 0); in Invoke()
222 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0); in Invoke()
225 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0); in Invoke()
226 execution.setInput(LSTMCell::kCellToForgetWeightsTensor, nullptr, 0); in Invoke()
227 execution.setInput(LSTMCell::kCellToOutputWeightsTensor, nullptr, 0); in Invoke()
232 execution.setInput(LSTMCell::kProjectionBiasTensor, nullptr, 0); in Invoke()
235 execution.setInput(LSTMCell::kProjectionWeightsTensor, nullptr, 0); in Invoke()
236 execution.setInput(LSTMCell::kProjectionBiasTensor, nullptr, 0); in Invoke()
239 ASSERT_EQ(execution.setInput(LSTMCell::kActivationParam, &activation_, sizeof(activation_)), in Invoke()
241 ASSERT_EQ(execution.setInput(LSTMCell::kCellClipParam, &cell_clip_, sizeof(cell_clip_)), in Invoke()
243 ASSERT_EQ(execution.setInput(LSTMCell::kProjClipParam, &proj_clip_, sizeof(proj_clip_)), in Invoke()
246 ASSERT_EQ(execution.compute(), Result::NO_ERROR); in Invoke()