#include <utility>
#include <vector>
-namespace nnc
+namespace mir_caffe
{
using namespace ::caffe;
{"Tile", CaffeOpType::tile},
{"WindowData", CaffeOpType::windowData}};
-} // namespace nnc
+} // namespace mir_caffe
* limitations under the License.
*/
-#ifndef NNCC_CAFFE_IMPORTER_H
-#define NNCC_CAFFE_IMPORTER_H
+#ifndef MIR_CAFFE_IMPORTER_H
+#define MIR_CAFFE_IMPORTER_H
#include <set>
#include <string>
#include "caffe_op_creator.h"
#include "caffe_op_types.h"
-namespace nnc
+namespace mir_caffe
{
class CaffeImporter
void processDeprecatedInput();
};
-} // namespace nnc
+} // namespace mir_caffe
-#endif // NNCC_CAFFE_IMPORTER_H
+#endif // MIR_CAFFE_IMPORTER_H
#include <iostream>
#include <set>
-namespace nnc
+namespace mir_caffe
{
static TensorVariant fixGroupedKernel(int groups, const TensorVariant &folded_kernel)
return {createOp<ops::ConcatOp>("", h_slices, 0)->getOutput(0)};
}
-} // namespace nnc
+} // namespace mir_caffe
* limitations under the License.
*/
-#ifndef NNCC_CAFFE_OP_CREATOR_H
-#define NNCC_CAFFE_OP_CREATOR_H
+#ifndef MIR_CAFFE_OP_CREATOR_H
+#define MIR_CAFFE_OP_CREATOR_H
#include <set>
#include <map>
#include "caffe/proto/caffe.pb.h"
-namespace nnc
+namespace mir_caffe
{
using mir::TensorVariant;
return _graph->create<OpType>(name, std::forward<Types>(args)...);
}
-} // namespace nnc
+} // namespace mir_caffe
-#endif // NNCC_CAFFE_OP_CREATOR_H
+#endif // MIR_CAFFE_OP_CREATOR_H
* limitations under the License.
*/
-#ifndef NNCC_CAFFE_OP_TYPES_H
-#define NNCC_CAFFE_OP_TYPES_H
+#ifndef MIR_CAFFE_OP_TYPES_H
+#define MIR_CAFFE_OP_TYPES_H
-namespace nnc
+namespace mir_caffe
{
enum class CaffeOpType
windowData
};
-} // namespace nnc
+} // namespace mir_caffe
-#endif // NNCC_CAFFE_OP_TYPES_H
+#endif // MIR_CAFFE_OP_TYPES_H
{
CaffeImporterPass::CaffeImporterPass(const std::string &filename)
- : _pimpl(new CaffeImporter(filename))
+ : _pimpl(new mir_caffe::CaffeImporter(filename))
{
}
#include <memory>
#include <string>
-namespace nnc
+namespace mir_caffe
{
class CaffeImporter;
+} // namespace mir_caffe
+
+namespace nnc
+{
class CaffeImporterPass : public NNImporter
{
~CaffeImporterPass() override;
private:
- std::unique_ptr<CaffeImporter> _pimpl;
+ std::unique_ptr<mir_caffe::CaffeImporter> _pimpl;
std::unique_ptr<mir::Graph> _graph;
};