tfkit::DecodeCommand Struct Reference

#include <DecodeCommand.hpp>

Collaboration diagram for tfkit::DecodeCommand:

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

Member Function Documentation

run()

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

overridevirtual

Implements cli::Command.

Definition at line 29 of file DecodeCommand.cpp.

{
  tensorflow::GraphDef graph_def;
 
  CmdArguments cmdargs(argc, argv);
 
  auto ioconfig = make_ioconfig(cmdargs);
 
  google::protobuf::io::IstreamInputStream is{ioconfig->in()};
  google::protobuf::io::CodedInputStream coded_is{&is};
 
  if (!graph_def.ParseFromCodedStream(&coded_is))
  {
    std::cerr << "ERROR: Failed to parse tensorflow model" << std::endl;
    return 255;
  }
 
  google::protobuf::io::OstreamOutputStream os{ioconfig->out()};
  google::protobuf::TextFormat::Print(graph_def, &os);
 
  return 0;
}

References tfkit::make_ioconfig().

---

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

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