337         NN_CHECK_EQ(aux_input_->shape().dimensions[0], input_->shape().dimensions[0]);  in Prepare()
338         NN_CHECK_EQ(aux_input_->shape().dimensions[1], input_->shape().dimensions[1]);  in Prepare()
383     const Shape& inputShape = input_->shape();  in Prepare()
420         *fwOutputActivationState = fw_activation_state_->shape();  in Prepare()
421         *fwOutputCellState = fw_cell_state_->shape();  in Prepare()
422         *bwOutputActivationState = bw_activation_state_->shape();  in Prepare()
423         *bwOutputCellState = bw_cell_state_->shape();  in Prepare()
443     std::vector<uint32_t> fw_output_dims = input_->shape().dimensions;  in Eval()
460             Shape bwInputShape = input_->shape();  in Eval()
464                 bwInputShape = aux_input_->shape();  in Eval()
477                         getNumberOfElements(fw_activation_state_->shape()));  in Eval()
478                 fw_output_cell_state.resize(getNumberOfElements(fw_cell_state_->shape()));  in Eval()
486                     params_, GetBuffer<const float>(input_), input_->shape(),  in Eval()
491                     fw_input_to_output_weights_->shape(),  in Eval()
496                     fw_recurrent_to_output_weights_->shape(),  in Eval()
529                         getNumberOfElements(bw_activation_state_->shape()));  in Eval()
530                 bw_output_cell_state.resize(getNumberOfElements(bw_cell_state_->shape()));  in Eval()
543                     bw_input_to_output_weights_->shape(),  in Eval()
548                     bw_recurrent_to_output_weights_->shape(),  in Eval()
583             Shape bwInputShape = input_->shape();  in Eval()
587                 bwInputShape = aux_input_->shape();  in Eval()
601                         getNumberOfElements(fw_activation_state_->shape()));  in Eval()
602                 fw_output_cell_state.resize(getNumberOfElements(fw_cell_state_->shape()));  in Eval()
610                     params_, GetBuffer<const _Float16>(input_), input_->shape(),  in Eval()
615                     fw_input_to_output_weights_->shape(),  in Eval()
620                     fw_recurrent_to_output_weights_->shape(),  in Eval()
654                         getNumberOfElements(bw_activation_state_->shape()));  in Eval()
655                 bw_output_cell_state.resize(getNumberOfElements(bw_cell_state_->shape()));  in Eval()
668                     bw_input_to_output_weights_->shape(),  in Eval()
673                     bw_recurrent_to_output_weights_->shape(),  in Eval()