Searched refs:input_weights_transposed (Results 1 – 2 of 2) sorted by relevance
/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/ |
H A D | CLQLSTMLayer.cpp | 710 …const TensorInfo input_weights_transposed(TensorShape(num_units, input_size), 1, input_to_forget_w… in validate() local 714 …COMPUTE_RETURN_ON_ERROR(CLTranspose::validate(input_to_forget_weights, &input_weights_transposed)); in validate() 715 …M_COMPUTE_RETURN_ON_ERROR(CLTranspose::validate(input_to_cell_weights, &input_weights_transposed)); in validate() 716 …COMPUTE_RETURN_ON_ERROR(CLTranspose::validate(input_to_output_weights, &input_weights_transposed)); in validate() 722 …N_ON_ERROR(CLTranspose::validate(lstm_params.input_to_input_weights(), &input_weights_transposed)); in validate() 744 …ARM_COMPUTE_RETURN_ON_ERROR(validate_mm(gemmlowp_info, input, &input_weights_transposed, &eff_bias… in validate() 779 …ARM_COMPUTE_RETURN_ON_ERROR(validate_mm(gemmlowp_info, input, &input_weights_transposed, &eff_bias… in validate() 815 …ARM_COMPUTE_RETURN_ON_ERROR(validate_mm(gemmlowp_info, input, &input_weights_transposed, &eff_bias… in validate() 854 …ARM_COMPUTE_RETURN_ON_ERROR(validate_mm(gemmlowp_info, input, &input_weights_transposed, &eff_bias… in validate()
|
/aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/ |
H A D | NEQLSTMLayer.cpp | 795 …const TensorInfo input_weights_transposed(TensorShape(num_units, input_size), 1, input_to_cell_wei… in validate() local 802 …M_COMPUTE_RETURN_ON_ERROR(NETranspose::validate(input_to_cell_weights, &input_weights_transposed)); in validate() 835 …ARM_COMPUTE_RETURN_ON_ERROR(validate_mm(gemmlowp_info, input, &input_weights_transposed, &eff_bias… in validate() 870 …ARM_COMPUTE_RETURN_ON_ERROR(validate_mm(gemmlowp_info, input, &input_weights_transposed, &eff_bias… in validate() 915 …ARM_COMPUTE_RETURN_ON_ERROR(validate_mm(gemmlowp_info, input, &input_weights_transposed, &eff_bias… in validate() 954 …ARM_COMPUTE_RETURN_ON_ERROR(validate_mm(gemmlowp_info, input, &input_weights_transposed, &eff_bias… in validate()
|