Searched refs:bw_cell_state_ (Results 1 – 2 of 2) sorted by relevance
146 bw_cell_state_ = GetInput(operation, operands, kBwInputCellStateTensor); in BidirectionalSequenceLSTM()422 *bwOutputCellState = bw_cell_state_->shape(); in Prepare()529 bw_output_cell_state.resize(getNumberOfElements(bw_cell_state_->shape())); in Eval()562 GetBuffer<const float>(bw_cell_state_), in Eval()654 bw_output_cell_state.resize(getNumberOfElements(bw_cell_state_->shape())); in Eval()687 GetBuffer<const _Float16>(bw_cell_state_), in Eval()
231 const RunTimeOperandInfo* bw_cell_state_; variable