Commit Graph

4 Commits (60ffc22026f43afd34a90b0b06d43fe7e0b2701b)

Author SHA1 Message Date
Leo Chen 60ffc22026
Refine bernoulli and unsqueeze op (#26842)
5 years ago
Yibing Liu 6e7bfe30a6
register fp16 kernel for some ops (#22650) (#22696)
5 years ago
liym27 84d221b667 improve unsqueeze op to support int, Tensor for argument axes (#20824)
5 years ago
zhongpu 5f627488db add kernel for unsqueeze_op and Add unsqueezed op test, test=develop (#19436)
6 years ago