tfkit::ConvertCommand Struct Reference

#include <ConvertCommand.hpp>

Collaboration diagram for tfkit::ConvertCommand:

Public Member Functions
int	run (int argc, const char *const *argv) const override
Public Member Functions inherited from cli::Command
virtual	~Command ()=default

Detailed Description

Definition at line 25 of file ConvertCommand.hpp.

Member Function Documentation

run()

int tfkit::ConvertCommand::run ( int  argc, const char *const *  argv  ) const

overridevirtual

Implements cli::Command.

Definition at line 99 of file ConvertCommand.cpp.

{
  tensorflow::GraphDef graph_def;
 
  // NOTE The current implementation accepts only command-line for the following form:
  //
  //   tfkit convert --input-format (pb or pbtxt) --output-format json ...
  //
  // TODO Support more options
  assert(argc >= 4);
  assert(std::string(argv[0]) == "--input-format");
  const std::string input_format{argv[1]};
  assert(std::string(argv[2]) == "--output-format");
  const std::string output_format{argv[3]};
 
  std::map<std::string, std::unique_ptr<Importer>> importers;
 
  importers["pb"] = std::make_unique<ImporterImpl<DataFormat::PBBIN>>();
  importers["pbtxt"] = std::make_unique<ImporterImpl<DataFormat::PBTXT>>();
 
  std::map<std::string, std::unique_ptr<Exporter>> exporters;
 
  exporters["json"] = std::make_unique<ExporterImpl<DataFormat::JSON>>();
 
  auto importer = importers.at(input_format).get();
  auto exporter = exporters.at(output_format).get();
 
  CmdArguments cmdargs(argc - 4, argv + 4);
 
  auto ioconfig = make_ioconfig(cmdargs);
 
  if (!importer->run(ioconfig->in(), graph_def))
  {
    std::cerr << "ERROR: Failed to import" << std::endl;
    return 255;
  }
 
  if (!exporter->run(graph_def, ioconfig->out()))
  {
    std::cerr << "ERROR: Failed to export" << std::endl;
    return 255;
  }
 
  return 0;
}

References tfkit::make_ioconfig().

Referenced by package.infer.session::inference().

---

The documentation for this struct was generated from the following files:

• compiler/tfkit/src/ConvertCommand.hpp
• compiler/tfkit/src/ConvertCommand.cpp