1. 28 2月, 2023 1 次提交
    • X
      Add flatten composite rule (#50672) · 8220771b
      xysheng-baidu 提交于
      * Add flatten composite rule
      
      * get the right xshape and pass func test
      
      * add cinn unit test
      
      * Remove cinn test, wait for it to be added after repair
      
      * add comp test to test_flatten_contiguous_range_op.py
      
      * remove func test on composite_ops
      
      * Add comments to maybe_wrap_dim func
      
      * remove commented code
      
      * fix the problem with 0D tensor case
      
      * add flatten split rule comment
      
      * fix syntax issues
      
      * block flatten on resnet_prim_cinn
      
      * remove maybe_wrap_dim func
      
      * Use none instead od xshape
      8220771b
  2. 24 2月, 2023 1 次提交
  3. 08 2月, 2023 1 次提交
  4. 07 2月, 2023 1 次提交
  5. 03 2月, 2023 1 次提交
  6. 02 2月, 2023 1 次提交
    • X
      【PRIM】Support use operator's output metadata info in constructing static... · d8643cb6
      Xiaoxu Chen 提交于
      【PRIM】Support use operator's output metadata info  in constructing static backward composite (#50043)
      
      * [prim] support custom target_gradients
      
      * support infershape after append one gradop
      
      * [prim] add simple net test
      
      * fix test_loop segment fault bug
      
      * [prim] fix infer shape segment fault bug when output of grad_op_desc is empty
      d8643cb6