Lines Matching refs:LSTMCell

195         ASSERT_EQ(execution.setInput(LSTMCell::k##X##Tensor, X##_.data(), \  in Invoke()
204         ASSERT_EQ(execution.setOutput(LSTMCell::k##X##Tensor, X##_.data(), \  in Invoke()
213             execution.setInput(LSTMCell::kInputToInputWeightsTensor, nullptr, 0);  in Invoke()
214             execution.setInput(LSTMCell::kRecurrentToInputWeightsTensor, nullptr, 0);  in Invoke()
219                 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0);  in Invoke()
222             execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0);  in Invoke()
223             execution.setInput(LSTMCell::kCellToForgetWeightsTensor, nullptr, 0);  in Invoke()
224             execution.setInput(LSTMCell::kCellToOutputWeightsTensor, nullptr, 0);  in Invoke()
229                 execution.setInput(LSTMCell::kProjectionBiasTensor, nullptr, 0);  in Invoke()
232             execution.setInput(LSTMCell::kProjectionWeightsTensor, nullptr, 0);  in Invoke()
233             execution.setInput(LSTMCell::kProjectionBiasTensor, nullptr, 0);  in Invoke()
236         ASSERT_EQ(execution.setInput(LSTMCell::kActivationParam,  in Invoke()
239         ASSERT_EQ(execution.setInput(LSTMCell::kCellClipParam,  in Invoke()
242         ASSERT_EQ(execution.setInput(LSTMCell::kProjClipParam,  in Invoke()