using namespace ::tflite;
-namespace nnc
+namespace mir_tflite
{
TfliteImporter::TfliteImporter(std::string filename) : _filename(std::move(filename))
}
}
-} // namespace nnc
+} // namespace mir_tflite
* limitations under the License.
*/
-#ifndef NNCC_TFLITE_IMPORTER_H
-#define NNCC_TFLITE_IMPORTER_H
+#ifndef MIR_TFLITE_IMPORTER_H
+#define MIR_TFLITE_IMPORTER_H
#include "schema_generated.h"
#include "mir/Graph.h"
#include <set>
#include <string>
-namespace nnc
+namespace mir_tflite
{
class TFLiteOpCreator;
mir::TensorVariant createTensor(const ::tflite::Tensor *t, const ::tflite::Buffer *b);
};
-} // namespace nnc
+} // namespace mir_tflite
-#endif // NNCC_TFLITE_IMPORTER_H
+#endif // MIR_TFLITE_IMPORTER_H
using namespace ::tflite;
-namespace nnc
+namespace mir_tflite
{
static void calculatePadding(tflite::Padding padding, const Shape &input_shape,
return {result->getOutput(0)};
}
-} // namespace nnc
+} // namespace mir_tflite
* limitations under the License.
*/
-#ifndef NNCC_TFLITE_OP_CREATOR_H
-#define NNCC_TFLITE_OP_CREATOR_H
+#ifndef MIR_TFLITE_OP_CREATOR_H
+#define MIR_TFLITE_OP_CREATOR_H
#include "schema_generated.h"
#include <set>
#include <vector>
-namespace nnc
+namespace mir_tflite
{
namespace ops = mir::ops;
return _graph->create<OpType>("", std::forward<Types>(args)...);
}
-} // namespace nnc
+} // namespace mir_tflite
-#endif // NNCC_TFLITE_OP_CREATOR_H
+#endif // MIR_TFLITE_OP_CREATOR_H
{
TfliteImporterPass::TfliteImporterPass(const std::string &filename)
- : _pimpl(new TfliteImporter(filename))
+ : _pimpl(new mir_tflite::TfliteImporter(filename))
{
}
#include <memory>
#include <string>
+namespace mir_tflite
+{
+class TfliteImporter;
+} // namespace mir_tflite
+
namespace nnc
{
-class TfliteImporter;
class TfliteImporterPass : public NNImporter
{
public:
~TfliteImporterPass() override;
private:
- std::unique_ptr<TfliteImporter> _pimpl;
+ std::unique_ptr<mir_tflite::TfliteImporter> _pimpl;
std::unique_ptr<mir::Graph> _graph;
};