Lines Matching refs:LSTMCell

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()