tfkit::EncodeCommand Struct Reference

#include <EncodeCommand.hpp>

Collaboration diagram for tfkit::EncodeCommand:

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

Member Function Documentation

run()

int EncodeCommand::run ( int  argc, const char *const *  argv  ) const

overridevirtual

Implements cli::Command.

Definition at line 29 of file EncodeCommand.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;
  }
 
  google::protobuf::io::OstreamOutputStream os{ioconfig->out()};
  google::protobuf::io::CodedOutputStream coded_os{&os};
 
  if (!graph_def.SerializeToCodedStream(&coded_os))
  {
    std::cerr << "ERROR: Failed to serialize" << 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/EncodeCommand.hpp
• compiler/tfkit/src/EncodeCommand.cpp