Added support for ArgMax in CNNPredictor
parent
6ccdf121
No related branches found
No related tags found
Showing
- src/main/resources/templates/gluon/CNNPredictor.ftl 1 addition, 1 deletionsrc/main/resources/templates/gluon/CNNPredictor.ftl
- src/main/resources/templates/gluon/execute.ftl 12 additions, 0 deletionssrc/main/resources/templates/gluon/execute.ftl
- src/test/resources/target_code/CNNPredictor_Alexnet.h 1 addition, 1 deletionsrc/test/resources/target_code/CNNPredictor_Alexnet.h
- src/test/resources/target_code/CNNPredictor_CifarClassifierNetwork.h 1 addition, 1 deletion...sources/target_code/CNNPredictor_CifarClassifierNetwork.h
- src/test/resources/target_code/CNNPredictor_VGG16.h 1 addition, 1 deletionsrc/test/resources/target_code/CNNPredictor_VGG16.h
Loading
Please register or sign in to comment