tflinspect::DumpOperatorVersion Class Reference

#include <Dump.h>

Collaboration diagram for tflinspect::DumpOperatorVersion:

Public Member Functions
	DumpOperatorVersion ()=default
void	run (std::ostream &os, const tflite::Model *model)
Public Member Functions inherited from tflinspect::DumpInterface
virtual	~DumpInterface ()=default

Detailed Description

Definition at line 54 of file Dump.h.

Constructor & Destructor Documentation

DumpOperatorVersion()

tflinspect::DumpOperatorVersion::DumpOperatorVersion ( )

default

Member Function Documentation

run()

void tflinspect::DumpOperatorVersion::run ( std::ostream &  os, const tflite::Model *  model  )

virtual

Implements tflinspect::DumpInterface.

Definition at line 148 of file Dump.cpp.

{
  std::map<std::string, int32_t> op_version_map;
 
  tflinspect::Reader reader(model);
 
  const uint32_t subgraph_size = reader.num_subgraph();
 
  for (uint32_t g = 0; g < subgraph_size; g++)
  {
    reader.select_subgraph(g);
    auto ops = reader.operators();
 
    // dump Conv2D, DepthwiseConv2D and its weight input operator
    for (uint32_t i = 0; i < ops->size(); ++i)
    {
      const auto op = ops->Get(i);
 
      auto op_name = reader.opcode_name(op);
      auto op_version = reader.opcodes().at(op->opcode_index())->version();
 
      if (op_version_map.find(op_name) == op_version_map.end() ||
          op_version_map[op_name] < op_version)
        op_version_map[op_name] = op_version;
    }
  }
 
  for (auto op : op_version_map)
  {
    os << op.first << "," << op.second << std::endl;
  }
}

References tflinspect::Reader::num_subgraph(), tflinspect::Reader::opcode_name(), tflinspect::Reader::opcodes(), tflinspect::Reader::operators(), and tflinspect::Reader::select_subgraph().

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

---

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

• compiler/tfl-inspect/src/Dump.h
• compiler/tfl-inspect/src/Dump.cpp