Commit Graph

193 Commits (4e1931481ab5ca69ef845144bf25b1283ce27c51)

Author SHA1 Message Date
yujianfeng 57cb1eeb14 Performance optimization of embedding_lookup
5 years ago
kingfo 09b437285b support mix precision in pynative
5 years ago
wangnan39@huawei.com 1cdf6a10ac rename operators of sparse optimizer
5 years ago
yujianfeng 188d74f15e Remove transdata and cast for internal outputs
5 years ago
yujianfeng 91306cbd34 Add ut for cpu kernel of sparse optimizer
5 years ago
mindspore-ci-bot 0a368494db !2499 HostAllGather and HostReduceScatter change to internal interface
5 years ago
Yi Huaijie 2eb739de6e change HostAllGather and HostReduceScatter to internal interface
5 years ago
yujianfeng 304427cd93 Add an output to apply_proximal_adagrad op register
5 years ago
baihuawei 862522376f add reduce and softmax_cross_entropy_with_logits
5 years ago
yujianfeng c3871d98cc Add implementation of SparseApplyProximalAdagrad cpu kernel
5 years ago
chenjianping 6034f9c1e2 support host reduce scatter and mpi config
5 years ago
buxue 66bbdb4a31 change tensor dtype and shape from function to attr
5 years ago
mindspore-ci-bot d6cc7089fc !1888 Add cpu kernel implement of sparse adam
5 years ago
yujianfeng c956dfff51 Add SparseAdam and SparseLazyAdam cpu kernel
5 years ago
baihuawei 9c74e39b12 update cpu lstm
5 years ago
yujianfeng 5d4b75838f Add SparseApplyFtrl cpu kernel
5 years ago
mindspore-ci-bot 063ad7c7f0 !1810 optimize slice/slicegrad performance
5 years ago
chenjianping af8108c9e1 support host reduce scatter and allgather
5 years ago
sunsuodong 78787931e1 optimize performance
5 years ago
mindspore-ci-bot ddfa1edefe !1618 fix lstm
5 years ago
baihuawei ea78e16e74 fix lstm
5 years ago
sunsuodong 63d9e47291 fix cpu StridedSliceGrad
5 years ago
liuwenhao4 7b911886ec Fixing some tiny faults about Pylint in my code(ops)
5 years ago
jinyaohui 86d197dfeb clean pylint
5 years ago
liuwenhao4 f87c70a9ab Fixing some tiny faults about Pylint in my code(ops)
5 years ago
mindspore-ci-bot bb72c0020b !1612 add cpu stridedslice
5 years ago
kswang f93e6beb52 add cpu strided slice
5 years ago
mindspore-ci-bot 14f9a6e31c !1310 add cpu lstm
5 years ago
mindspore-ci-bot 1bfa0dc5d5 !1265 add cpu transpose
5 years ago
liuwenhao4 a7ad0d0a49 Fixing some tiny faults about Pylint in my code(ops)
5 years ago
baihuawei 9bcdf4cbdc add lstm
5 years ago
sunsuodong 5cf29f2478 multi_input
5 years ago
baihuawei eb0f897d31 add cpu transpose
5 years ago
sunsuodong ade90be427 addn
5 years ago
liuwenhao4 f3f0cbaeee Fixing some tiny faults about Pylint in my code(ops)
5 years ago
jinyaohui fbdba6e4da clean pylint
5 years ago
mindspore-ci-bot 6ffb3c97cd !1058 add host cpu kernel: GatherV2/Concat/Slice/SliceGrad
5 years ago
jinyaohui 5a914994ba clean pylint
5 years ago
sunsuodong 28241d0293 lstm ops
5 years ago
jinyaohui bcfaff97f9 clean pylint
5 years ago
kswang 6775190e48 add cpu one hot
5 years ago
wangnan39@huawei.com 2604acedcb extend conv stride and dilation to 2d
5 years ago
zhunaipan 930a1fb0a8 initial version
5 years ago