Commit Graph

5 Commits (c21a979790aebefffdde5c470dc406a2a81959e7)

Author SHA1 Message Date
joejiong 1a532d5133
add uint8 support for squeeze operator (#28734)
4 years ago
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