[nnc] Initial implementation of caffe2_op_creator (#2333)
authorIvan Vagin/AI Tools Lab /SRR/Engineer/삼성전자 <ivan.vagin@samsung.com>
Fri, 7 Dec 2018 11:23:02 +0000 (14:23 +0300)
committerEfimov Alexander/AI Tools Lab/./Samsung Electronics <a.efimov@samsung.com>
Fri, 7 Dec 2018 11:23:02 +0000 (14:23 +0300)
commit7b84a6938a383f0dc0905decec8ede6c7c591af3
tree640180f581d5d9bc25339b50e86c039946c07950
parentdb752891abf9268c99e7171147d46e78b683ff0a
[nnc] Initial implementation of caffe2_op_creator (#2333)

Initial implementation of caffe2_op_creator.

`mobilenet` supported, to support `inception` model - need to support custom paddings in pooling ops and test not tested operation conversations.

Implemented ops:
- Add
- AveragePool
- Conv
- Concat
- Dropout
- FC
- GivenTensorFill
- MaxPool
- Mul
- Relu
- Softmax
- SpatialBN
- Sum

Not tested ops:
- Add
- Concat
- Mul
- SpatialBN

Signed-off-by: Ivan Vagin <ivan.vagin@samsung.com>
12 files changed:
contrib/nnc/driver/Options.cpp
contrib/nnc/include/passes/caffe2_frontend/caffe2_importer.h
contrib/nnc/include/passes/common_frontend/op_creator_helper.h [new file with mode: 0644]
contrib/nnc/passes/caffe2_frontend/caffe2_importer.cpp
contrib/nnc/passes/caffe2_frontend/caffe2_op_creator.cpp [new file with mode: 0644]
contrib/nnc/passes/caffe2_frontend/caffe2_op_creator.h [new file with mode: 0644]
contrib/nnc/passes/caffe2_frontend/caffe2_op_types.h
contrib/nnc/passes/caffe2_frontend/caffe2_proto_helper.cpp
contrib/nnc/passes/caffe2_frontend/caffe2_proto_helper.h
contrib/nnc/passes/caffe_frontend/caffe_op_creator.cpp
contrib/nnc/passes/common_frontend/CMakeLists.txt
contrib/nnc/passes/common_frontend/op_creator_helper.cpp [new file with mode: 0644]