Added axis param for TopK

VeriSilicon / TIM-VX

VeriSilicon Tensor Interface Module

Other

216 stars 85 forks source link

Closed chenfeiyue-cfy closed 1 year ago

chenfeiyue-cfy commented 1 year ago

Topk support specifying dimensions with later internal ovxlib Need internal ovxlib update

Type: New Feature

sunshinemyson commented 1 year ago

You can rebase this now

chenfeiyue-cfy commented 1 year ago

You can rebase this now

Done