search

microsoft / nnfusion

A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.
MIT License
937 stars 157 forks source link

fix argmax/argmin missing arguments in tensorflow import unitest #501

Closed LeiWang1999 closed 1 year ago

LeiWang1999 commented 1 year ago
  1. fix argmax/argmin missing arguments in tensorflow import unitest.
  2. fix typo in src/nnfusion/core/operators/generic_op/generic_op_define/ArgMin.cpp