circleinspect::DumpTensorShape Class Reference

#include <Dump.h>

Collaboration diagram for circleinspect::DumpTensorShape:

Public Member Functions
	DumpTensorShape ()=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 81 of file Dump.h.

Constructor & Destructor Documentation

DumpTensorShape()

circleinspect::DumpTensorShape::DumpTensorShape ( )

default

Member Function Documentation

run()

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

virtual

Implements circleinspect::DumpInterface.

Definition at line 247 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 tensors = reader.tensors();
 
    for (uint32_t i = 0; i < tensors->size(); ++i)
    {
      const auto tensor = tensors->Get(i);
      auto shape = tensor->shape_signature() ? tensor->shape_signature() : tensor->shape();
      os << reader.tensor_name(tensor) << " [";
      for (uint32_t i = 0; i < shape->size(); i++)
      {
        os << shape->Get(i);
        if (i != shape->size() - 1)
        {
          os << ",";
        }
      }
      os << "]" << std::endl;
    }
  }
}

References mio::circle::Reader::num_subgraph(), mio::circle::Reader::select_subgraph(), mio::circle::Reader::tensor_name(), and mio::circle::Reader::tensors().

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