- 28 2月, 2023 1 次提交
-
-
由 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
-
- 24 2月, 2023 1 次提交
-
-
由 WangZhen 提交于
* Add bert prim and cinn test
-
- 08 2月, 2023 1 次提交
-
-
由 WangZhen 提交于
-
- 07 2月, 2023 1 次提交
-
-
由 cyber-pioneer 提交于
move composite test case remove unuseful var add composite op blacklist
-
- 03 2月, 2023 1 次提交
-
-
由 WangZhen 提交于
Add test_resnet for prim and CINN
-
- 02 2月, 2023 1 次提交
-
-
由 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
-