return convert_shape(shape_def);
}
-void add_customop(tf2tflite::CustomOpInfoDef &def, moco::tf::ModelSignature &sig)
+void add_customop(tf2tflite::CustomOpInfoDef &def, moco::ModelSignature &sig)
{
for (const auto &custom_op : def.custom_op())
{
namespace tf2tflite
{
-void load_customop_conf(const std::string &path, moco::tf::ModelSignature &sig)
+void load_customop_conf(const std::string &path, moco::ModelSignature &sig)
{
CustomOpInfoDef def;
{
/// @brief Loads customop.conf into ModelSignature
-void load_customop_conf(const std::string &path, moco::tf::ModelSignature &sig);
+void load_customop_conf(const std::string &path, moco::ModelSignature &sig);
} // namespace tf2tflite
namespace
{
-std::unique_ptr<loco::Graph> import(const moco::tf::ModelSignature &sig, const std::string &path)
+std::unique_ptr<loco::Graph> import(const moco::ModelSignature &sig, const std::string &path)
{
moco::tf::Frontend frontend;
return frontend.load(sig, path.c_str(), moco::tf::Frontend::FileType::Binary);
std::cout << "Read '" << info_path << "'" << std::endl;
- moco::tf::ModelSignature sig;
+ moco::ModelSignature sig;
{
for (const auto &info : nnkit::support::tftestinfo::parse(info_path.c_str()))
{
switch (info->kind())
{
case nnkit::support::tftestinfo::ParsedTensor::Kind::Input:
- sig.add_input(moco::tf::TensorName{info->name()});
+ sig.add_input(moco::TensorName{info->name()});
sig.shape(info->name(), info->shape());
break;
case nnkit::support::tftestinfo::ParsedTensor::Kind::Output:
- sig.add_output(moco::tf::TensorName{info->name()});
+ sig.add_output(moco::TensorName{info->name()});
sig.shape(info->name(), info->shape());
break;