"src/git@developer.sourcefind.cn:lacacy/qwen_lmdeploy.git" did not exist on "0cc9d095af5ce5339f5bb563435f4c6183f787ee"
  • Lei Wang's avatar
    [Layout] Support layout forward with multi dimension (#867) · 9cbbbbc6
    Lei Wang authored
    * Enhance LayoutNode::Forward method to handle variable transformations more robustly
    
    - Updated the method to check for a minimum number of input dimensions.
    - Introduced a mechanism to transform the last InputDim() elements of the input variables.
    - Concatenated transformed variables with the remaining input variables for a comprehensive output.
    
    * Refactor LayoutNode::Forward method for improved readability
    
    - Removed unnecessary whitespace to enhance code clarity.
    - Maintained existing functionality while streamlining the transformation process of input variables.
    9cbbbbc6
layout.cc 17.6 KB