InitCommand Struct Reference

#include <InitCommand.h>

Collaboration diagram for InitCommand:

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 22 of file InitCommand.h.

Member Function Documentation

run()

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

overridevirtual

Implements cli::Command.

Definition at line 29 of file InitCommand.cpp.

{
  // Read prototxt from standard input
  ::caffe::NetParameter in;
  {
    google::protobuf::io::IstreamInputStream is{&std::cin};
    if (!google::protobuf::TextFormat::Parse(&is, &in))
    {
      std::cerr << "ERROR: Failed to parse prototxt" << std::endl;
      return 255;
    }
  }
 
  // Upgrade prototxt if necessary
  if (::caffe::NetNeedsUpgrade(in))
  {
    if (!::caffe::UpgradeNetAsNeeded("<stdin>", &in))
    {
      std::cerr << "ERROR: Failed to upgrade prototxt" << std::endl;
      return 255;
    }
  }
 
  ::caffe::Net<float> net(in);
 
  // Extract initialized parameters
  ::caffe::NetParameter out;
  {
    net.ToProto(&out);
  }
 
  // Write initialized parameters to standard output
  google::protobuf::io::OstreamOutputStream os{&std::cout};
  google::protobuf::TextFormat::Print(out, &os);
 
  return 0;
}

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

---

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

• compiler/caffegen/src/InitCommand.h
• compiler/caffegen/src/InitCommand.cpp