Lines Matching refs:getNumberOfElements
476 getNumberOfElements(fw_activation_state_->shape())); in Eval()
477 fw_output_cell_state.resize(getNumberOfElements(fw_cell_state_->shape())); in Eval()
482 std::vector<float> fw_scratch_buffer(getNumberOfElements(fw_scratch_shape_)); in Eval()
528 getNumberOfElements(bw_activation_state_->shape())); in Eval()
529 bw_output_cell_state.resize(getNumberOfElements(bw_cell_state_->shape())); in Eval()
534 std::vector<float> bw_scratch_buffer(getNumberOfElements(bw_scratch_shape_)); in Eval()
600 getNumberOfElements(fw_activation_state_->shape())); in Eval()
601 fw_output_cell_state.resize(getNumberOfElements(fw_cell_state_->shape())); in Eval()
606 std::vector<_Float16> fw_scratch_buffer(getNumberOfElements(fw_scratch_shape_)); in Eval()
653 getNumberOfElements(bw_activation_state_->shape())); in Eval()
654 bw_output_cell_state.resize(getNumberOfElements(bw_cell_state_->shape())); in Eval()
659 std::vector<_Float16> bw_scratch_buffer(getNumberOfElements(bw_scratch_shape_)); in Eval()