Lines Matching refs:numCells
437 const uint32_t numCells = getSizeOfDimension(input_to_output_weights_shape, 0); in LSTMEvalFloat32() local
472 cell_state_in_buffer + batchSize * numCells); in LSTMEvalFloat32()
509 cell_state_out_buffer + batchSize * numCells); in LSTMEvalFloat32()
557 const uint32_t numCells = getSizeOfDimension(input_to_output_weights_shape, 0); in LSTMEvalFloat16() local
567 std::vector<float> input_to_input_weights_float32(numCells * inputSize); in LSTMEvalFloat16()
571 std::vector<float> input_to_forget_weights_float32(numCells * inputSize); in LSTMEvalFloat16()
573 std::vector<float> input_to_cell_weights_float32(numCells * inputSize); in LSTMEvalFloat16()
575 std::vector<float> input_to_output_weights_float32(numCells * inputSize); in LSTMEvalFloat16()
578 std::vector<float> recurrent_to_input_weights_float32(numCells * outputSize); in LSTMEvalFloat16()
583 std::vector<float> recurrent_to_forget_weights_float32(numCells * outputSize); in LSTMEvalFloat16()
586 std::vector<float> recurrent_to_cell_weights_float32(numCells * outputSize); in LSTMEvalFloat16()
588 std::vector<float> recurrent_to_output_weights_float32(numCells * outputSize); in LSTMEvalFloat16()
592 std::vector<float> cell_to_input_weights_float32(numCells); in LSTMEvalFloat16()
596 std::vector<float> cell_to_forget_weights_float32(numCells); in LSTMEvalFloat16()
600 std::vector<float> cell_to_output_weights_float32(numCells); in LSTMEvalFloat16()
609 std::vector<float> aux_input_to_input_weights_float32(numCells * inputSize); in LSTMEvalFloat16()
614 std::vector<float> aux_input_to_forget_weights_float32(numCells * inputSize); in LSTMEvalFloat16()
619 std::vector<float> aux_input_to_cell_weights_float32(numCells * inputSize); in LSTMEvalFloat16()
624 std::vector<float> aux_input_to_output_weights_float32(numCells * inputSize); in LSTMEvalFloat16()
630 std::vector<float> input_gate_bias_float32(numCells); in LSTMEvalFloat16()
634 std::vector<float> forget_gate_bias_float32(numCells); in LSTMEvalFloat16()
636 std::vector<float> cell_bias_float32(numCells); in LSTMEvalFloat16()
638 std::vector<float> output_gate_bias_float32(numCells); in LSTMEvalFloat16()
641 std::vector<float> projection_weights_float32(numCells * outputSize); in LSTMEvalFloat16()
650 std::vector<float> input_layer_norm_weights_float32(numCells); in LSTMEvalFloat16()
654 std::vector<float> forget_layer_norm_weights_float32(numCells); in LSTMEvalFloat16()
659 std::vector<float> cell_layer_norm_weights_float32(numCells); in LSTMEvalFloat16()
663 std::vector<float> output_layer_norm_weights_float32(numCells); in LSTMEvalFloat16()
671 std::vector<float> cell_state_out_float32(batchSize * numCells); in LSTMEvalFloat16()
676 std::vector<float> scratch_buffer_float32(params.use_cifg ? 3 * batchSize * numCells in LSTMEvalFloat16()
677 : 4 * batchSize * numCells); in LSTMEvalFloat16()
708 std::vector<float> cellStateInCurrentTimeStep(batchSize * numCells); in LSTMEvalFloat16()