Commit Graph

390 Commits

Author SHA1 Message Date
z00512249 3470565532 add selu kernerl cpu implement 2022-04-09 14:57:27 +08:00
z00512249 e50a7bfe41 add lerp kernerl cpu implement 2022-04-09 09:57:10 +08:00
hezhenhao1 54d290a813 Add cpu implementation of IsClose operator. 2022-04-08 12:44:48 +08:00
hezhenhao1 33beadec59 Add CPU implementation of ScatterNdMul operator. 2022-04-07 09:13:00 +08:00
wangyanling 7ce37d2990 restore test case bug of broadcastto op 2022-03-23 14:33:21 +08:00
i-robot a3a08693a9
!31672 add maskedfill op for cpu
Merge pull request !31672 from 范吉斌/maskedfill
2022-03-22 11:31:31 +00:00
i-robot 374d26668f
!31671 add intopk op for cpu
Merge pull request !31671 from 范吉斌/intopk
2022-03-22 11:30:05 +00:00
fan-jibin 70372f8297 add intopk op for cpu 2022-03-22 11:48:06 +08:00
fan-jibin 4e4873d734 add maskedfill op for cpu 2022-03-22 11:45:14 +08:00
buxue 66eb76476d develop nllloss and nlllossgrad for CPU and lite 2022-03-22 10:35:07 +08:00
i-robot 111289bfdc
!31302 disable testcase on macos
Merge pull request !31302 from chenweifeng/disable-testcase-on-mac
2022-03-16 08:55:30 +00:00
wilfChen b5f5e0e632 disable testcase on mac 2022-03-16 14:32:28 +08:00
buxue b7f90e2b16 implement Conv2dBackward and log of pool 2022-03-15 21:26:40 +08:00
wilfChen 002ef3bb1f priority replay buffer test case 2022-03-10 09:01:54 +08:00
yanghaoran 5352e13329 takedown testcases to ensure gate stability 2022-03-09 10:18:51 +08:00
buxue 50f3ad00ab develop maxpool3d, avgpool3d and conv3d on CPU(including forward and backward) 2022-03-07 20:44:38 +08:00
jiangzhenguang 64da650786 add half pixel center to resize bilinear 2022-03-05 15:43:20 +08:00
VectorSL 8b45b00f7c update tensorarry stack for static usage 2022-02-14 09:14:35 +08:00
zuochuanyong 9aa370ddda onednn dfx testcase skip win and mac 2022-02-10 11:07:43 +08:00
hezhenhao1 f08b7a53b9 Fix CI probabilistic filter failed in gate building<test_arithmetic_op.py> 2022-01-25 20:45:26 +08:00
yanghaoran 27d28f3d45 takedown test_broadcast_grade_cpu 2022-01-25 09:46:00 +08:00
z00512249 7257cf129a add complex functions for cpu backend. 2022-01-21 15:59:58 +08:00
i-robot 3b011e6895
!26009 [feat][assistant][I48O6D] Add Square operator
Merge pull request !26009 from 李定维/Square
2022-01-17 08:58:00 +00:00
zuochuanyong 9a1f9d69f6 onednn dfx 2022-01-14 20:30:53 +08:00
ldv 630aea79cb [feat][assistant][I48O6D] Add Square operator 2022-01-14 15:31:28 +08:00
i-robot 7384cb2689
!28546 fused global norm for cpu adafactor
Merge pull request !28546 from kisnwang/add-cpu-adafactor
2022-01-11 12:22:49 +00:00
kswang 643b7b7ce4 fused global norm for cpu fused adafactor 2022-01-11 14:56:48 +08:00
i-robot 221246711a
!28330 functional select support x or y is a scalar
Merge pull request !28330 from wangnan39/support_functional_select
2022-01-05 07:13:49 +00:00
王南 eab11f8677 functional select support x or y is a scalar 2022-01-05 10:12:44 +08:00
kswang 1b37d4302d add fused adafactor primitive 2021-12-29 17:12:34 +08:00
i-robot d12b05f1f8 !27949 fix mac compile and fix st
Merge pull request !27949 from xulei/fix_mac
2021-12-22 01:16:49 +00:00
zhujingxuan ff13ddd353 reduce testcase 2021-12-21 17:37:21 +08:00
xulei 683a159e72 fix mac compile and fix st 2021-12-21 09:23:33 +08:00
i-robot deea2c9b6b !26942 unify GRU and LSTM operaters and change LSTMCell to right version
Merge pull request !26942 from 吕昱峰(Nate.River)/rnn_op
2021-12-14 16:12:06 +00:00
i-robot b97df28d6c !27581 Add bool for tensorarray
Merge pull request !27581 from VectorSL/add-bool-for-tensorarray
2021-12-14 02:48:32 +00:00
VectorSL 91766dbcd8 add bool for tesorarray 2021-12-13 15:07:51 +08:00
huangbo77 4300938799 Modifies test case for RandomChoiceWIthMask on mac 2021-12-09 11:27:24 +08:00
lvyufeng 7de53ec3fe unify GRU and LSTM operaters and change LSTMCell to right version 2021-12-08 16:51:51 +08:00
z00512249 a1b8c2f065 while inplace op for cpu backend 2021-12-06 17:41:40 +08:00
i-robot 84d2e54871 !26959 TensorArray CPU
Merge pull request !26959 from VectorSL/tensor-array-cpu
2021-12-01 03:31:26 +00:00
VectorSL cb3d25c8f0 add cpu tensor array 2021-11-30 22:35:24 +08:00
zhujingxuan a18b9e703e move testcases 2021-11-30 10:52:02 +08:00
z00512249 872d8b37da fix lu factor and solve for cpu backend 2021-11-26 16:24:01 +08:00
wenbean 13409f519f Unify GPU/CPU ops input/output(col/rolmajor), modify related testcases, add linalg function and testcases 2021-11-19 14:40:27 +08:00
wenbean 9b305b231d Add GPU eigenvalues/eigenvector for symmetric mtrix(real and complex)
add cuda ops real to complex(imag is 0j)
cpu/gpu eigh ops alreas returns complex eigenvalues for complex input
retuns real eigenvalues for real input
2021-11-18 11:23:36 +08:00
z00512249 36032e7ee2 add cholesky, cho_factor primitive and backend gpu implements 2021-11-15 21:19:56 +08:00
hezhenhao1 4c12385050 Fix bug: AbsGrad support int as input. 2021-11-12 10:46:45 +08:00
i-robot 3caa6cb09e !25898 split generic matrix/symmetric matrix eigenvalues to eig/eigh(same with scipy)
Merge pull request !25898 from wuwenbing/master
2021-11-05 08:37:25 +00:00
i-robot fb155743a7 !25921 Fix: cpu real_div support fp16.
Merge pull request !25921 from hezhenhao1/add_fp16
2021-11-05 07:10:51 +00:00
i-robot b6833ec3f2 !25872 add solve_triangular
Merge pull request !25872 from zhujingxuan/solve_triangular
2021-11-05 06:15:35 +00:00