circleinspect::DumpConv2DWeight Class Reference

#include <Dump.h>

Collaboration diagram for circleinspect::DumpConv2DWeight:

Public Member Functions
	DumpConv2DWeight ()=default
void	run (std::ostream &os, const circle::Model *model, const std::vector< char > *data)
Public Member Functions inherited from circleinspect::DumpInterface
virtual	~DumpInterface ()=default

Detailed Description

Definition at line 45 of file Dump.h.

Constructor & Destructor Documentation

DumpConv2DWeight()

circleinspect::DumpConv2DWeight::DumpConv2DWeight ( )

default

Member Function Documentation

run()

void circleinspect::DumpConv2DWeight::run ( std::ostream &  os, const circle::Model *  model, const std::vector< char > *  data  )

virtual

Implements circleinspect::DumpInterface.

Definition at line 97 of file Dump.cpp.

{
  mio::circle::Reader reader(model, data);
 
  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 bc = reader.builtin_code(op);
 
      if (bc == circle::BuiltinOperator_CONV_2D || bc == circle::BuiltinOperator_DEPTHWISE_CONV_2D)
      {
        const std::vector<int32_t> &inputs = mio::circle::as_index_vector(op->inputs());
        if (inputs.size() < 2)
        {
          throw std::runtime_error("Operator has invalid input");
        }
        auto weight_input = inputs[1]; // Tensor ID of weight input
 
        const auto op_weight = operator_match_output(reader, weight_input);
        const auto buffer_size = tensor_buffer_size(reader, weight_input);
 
        std::string weight_op_name = "?";
 
        if (op_weight == nullptr && buffer_size > 0)
        {
          weight_op_name = "CONST";
        }
        else if (op_weight != nullptr)
        {
          weight_op_name = reader.opcode_name(op_weight);
        }
 
        auto op_name = reader.opcode_name(op);
        os << op_name << "," << weight_op_name << std::endl;
      }
    }
  }
}

References mio::circle::as_index_vector(), mio::circle::Reader::builtin_code(), mio::circle::Reader::num_subgraph(), mio::circle::Reader::opcode_name(), mio::circle::Reader::operators(), and mio::circle::Reader::select_subgraph().

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

---

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

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