Lines Matching refs:execution
208 Execution execution(&compilation); in Invoke() local
211 execution.setInput(LSTMCell::k##X##Tensor, X##_.data(), sizeof(float) * X##_.size()), \ in Invoke()
221 execution.setOutput(LSTMCell::k##X##Tensor, X##_.data(), sizeof(float) * X##_.size()), \ in Invoke()
229 execution.setInput(LSTMCell::kInputToInputWeightsTensor, nullptr, 0); in Invoke()
230 execution.setInput(LSTMCell::kRecurrentToInputWeightsTensor, nullptr, 0); in Invoke()
235 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0); in Invoke()
238 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0); in Invoke()
239 execution.setInput(LSTMCell::kCellToForgetWeightsTensor, nullptr, 0); in Invoke()
240 execution.setInput(LSTMCell::kCellToOutputWeightsTensor, nullptr, 0); in Invoke()
245 execution.setInput(LSTMCell::kProjectionBiasTensor, nullptr, 0); in Invoke()
248 execution.setInput(LSTMCell::kProjectionWeightsTensor, nullptr, 0); in Invoke()
249 execution.setInput(LSTMCell::kProjectionBiasTensor, nullptr, 0); in Invoke()
252 ASSERT_EQ(execution.setInput(LSTMCell::kActivationParam, &activation_, sizeof(activation_)), in Invoke()
254 ASSERT_EQ(execution.setInput(LSTMCell::kCellClipParam, &cell_clip_, sizeof(cell_clip_)), in Invoke()
256 ASSERT_EQ(execution.setInput(LSTMCell::kProjClipParam, &proj_clip_, sizeof(proj_clip_)), in Invoke()
259 ASSERT_EQ(execution.compute(), Result::NO_ERROR); in Invoke()