- 24 3月, 2021 2 次提交
- 23 3月, 2021 5 次提交
-
-
由 Leo Chen 提交于
* add npu kernel for concat op * add npu kernel for concat op * refine code * update * refine concat_grad
-
由 Leo Chen 提交于
* support list of tensor as npu input * add comment * fix typo * fix typo
-
由 Leo Chen 提交于
-
由 lw921014 提交于
add c_reduce_sum op
-
由 lilong12 提交于
Add 3d Parallelism Co-authored-by: NWangXi <wangxi16@baidu.com> Co-authored-by: NJZ-LIANG <jianzhongliang10@gmail.com> Co-authored-by: Nroot <root@yq01-sys-hic-k8s-v100-box-a225-0562.yq01.baidu.com>
-
- 22 3月, 2021 1 次提交
-
-
由 xiayanming 提交于
* fix gather grad kernel diff * fix gather grad kernel diff * fix gather review bug
-
- 20 3月, 2021 1 次提交
-
-
由 oyjxer 提交于
-
- 19 3月, 2021 3 次提交
- 18 3月, 2021 4 次提交
- 17 3月, 2021 6 次提交
-
-
由 liym27 提交于
-
由 OleNet 提交于
-
由 Meiyim 提交于
-
由 oyxuan-11 提交于
-
由 OleNet 提交于
* add topk op * add cmake * update topk npu op * refactor func * fix test not go npu TopKD bug * NPUPlace(4) to NPUPlace(0) * update comment Co-authored-by: Noyjxer <1728722986@qq.com>
-
由 OleNet 提交于
* add increment * fix * update test increment op inplace * update increment op * increment b = 2 Co-authored-by: Noyjxer <1728722986@qq.com>
-
- 16 3月, 2021 8 次提交
-
-
由 Leo Chen 提交于
* fix shape of dx * refine code
-
由 Meiyim 提交于
* [npu] support npu kernel for `expand`
-
由 pangyoki 提交于
* update unittest * add TensorCopy in npu grad kernel
-
由 OleNet 提交于
* update mean op * update mean op * give a better test activation Co-authored-by: Noyjxer <1728722986@qq.com>
-
由 Leo Chen 提交于
* init * fix bugs
-
由 zhang wenhui 提交于
* add mul * add test mul
-
由 Leo Chen 提交于
* add npu kernel for adam * refine code * disable test * modify atol
-
- 15 3月, 2021 10 次提交
-
-
由 pangyoki 提交于
* add reduce_sum * fix broadcastd * fix test * fix * add unsqueeze in reduce_sum * add template * add unittest for keep_dim * test reduce_all Co-authored-by: Nfrankwhzhang <frankwhzhang@126.com>
-
由 Leo Chen 提交于
-
由 oyxuan-11 提交于
Co-authored-by: Nfrankwhzhang <frankwhzhang@126.com>
-
由 Leo Chen 提交于
-
由 oyxuan-11 提交于
* Support npu kernel scatter op * Add more test
-
由 Meiyim 提交于
* [npu] support npu kernel for `less than` * remove int* kernel * cleanup
-
由 Meiyim 提交于
* [npu] support npu kernel `table_lookup_v2` * clean up * +python test * +cmake * clean up * remove int8 kernel + python unitest for fp16 * clean up
-
由 oyjxer 提交于
-
由 oyjxer 提交于
-
由 oyjxer 提交于
-