Commit Graph

12 Commits

Author SHA1 Message Date
huangbingjian 66309632c1 fix the negative axis problem of reduce_eliminate 2021-11-30 13:11:01 +08:00
hezhenhao1 cb13783059 Add fp64 as input type fot GPU op of ReduceMin and ReduceMean. 2021-11-08 17:15:32 +08:00
huangbingjian ad8f73f32b set default context mode to GRAPH_MODE 2021-06-09 14:40:49 +08:00
yujianfeng 41189781f3 support scalar input for cell 2021-02-25 09:56:15 +08:00
TFBunny 6cd7dc42e9 add testcases and dynamic shape to reduce ops 2021-01-27 12:26:22 -05:00
TFbunny d0b2f17445 add dynamic shape to GPU ReduceMean with testcases 2021-01-12 16:19:01 -05:00
linqingke fb405ee6f4 broadcast, slice, scatter_nd ops optimizer. 2020-08-08 10:30:05 +08:00
buxue 66bbdb4a31 change tensor dtype and shape from function to attr 2020-06-12 19:03:23 +08:00
cristoval f6c20178d2 fix pylint check issues 2020-05-25 15:34:01 +08:00
jinyaohui 5a914994ba clean pylint 2020-05-18 16:42:35 +08:00
ZPaC d3936b9f2a GPU kernels adapt with special dimensions. 2020-05-08 10:26:36 +08:00
zhunaipan 930a1fb0a8 initial version
Signed-off-by: leonwanghui <leon.wanghui@huawei.com>
2020-03-27 22:54:54 +08:00