tfkit::PackCommand Struct Reference

#include <PackCommand.hpp>

Collaboration diagram for tfkit::PackCommand:

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 PackCommand.hpp.

Member Function Documentation

run()

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

overridevirtual

Implements cli::Command.

Definition at line 145 of file PackCommand.cpp.

{
  tensorflow::GraphDef graph_def;
 
  CmdArguments cmdargs(argc, argv);
 
  auto ioconfig = make_ioconfig(cmdargs);
 
  google::protobuf::io::IstreamInputStream is{ioconfig->in()};
 
  if (!google::protobuf::TextFormat::Parse(&is, &graph_def))
  {
    std::cerr << "ERROR: Failed to parse prototxt" << std::endl;
    return 255;
  }
 
  // convert float_val to tensor_content
  pack(graph_def);
 
  google::protobuf::io::OstreamOutputStream os{ioconfig->out()};
  google::protobuf::TextFormat::Print(graph_def, &os);
 
  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/PackCommand.hpp
• compiler/tfkit/src/PackCommand.cpp