Lines Matching refs:execution
207 Execution execution(&compilation); in Invoke() local
210 execution.setInput(LSTMCell::k##X##Tensor, X##_.data(), sizeof(float) * X##_.size()), \ in Invoke()
220 execution.setOutput(LSTMCell::k##X##Tensor, X##_.data(), sizeof(float) * X##_.size()), \ in Invoke()
228 execution.setInput(LSTMCell::kInputToInputWeightsTensor, nullptr, 0); in Invoke()
229 execution.setInput(LSTMCell::kRecurrentToInputWeightsTensor, nullptr, 0); in Invoke()
234 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0); in Invoke()
237 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0); in Invoke()
238 execution.setInput(LSTMCell::kCellToForgetWeightsTensor, nullptr, 0); in Invoke()
239 execution.setInput(LSTMCell::kCellToOutputWeightsTensor, nullptr, 0); in Invoke()
244 execution.setInput(LSTMCell::kProjectionBiasTensor, nullptr, 0); in Invoke()
247 execution.setInput(LSTMCell::kProjectionWeightsTensor, nullptr, 0); in Invoke()
248 execution.setInput(LSTMCell::kProjectionBiasTensor, nullptr, 0); in Invoke()
251 ASSERT_EQ(execution.setInput(LSTMCell::kActivationParam, &activation_, sizeof(activation_)), in Invoke()
253 ASSERT_EQ(execution.setInput(LSTMCell::kCellClipParam, &cell_clip_, sizeof(cell_clip_)), in Invoke()
255 ASSERT_EQ(execution.setInput(LSTMCell::kProjClipParam, &proj_clip_, sizeof(proj_clip_)), in Invoke()
258 ASSERT_EQ(execution.compute(), Result::NO_ERROR); in Invoke()