circle_resizer::ModelEditor Class Reference

#include <ModelEditor.h>

Public Member Functions
	ModelEditor (std::shared_ptr< CircleModel > circle_model)
	Initialize the editor with CircleModel object.
ModelEditor &	resize_inputs (const std::vector< Shape > &new_inputs_shapes)
	Resize the model. It means changing shape of the inputs and propagating changes through the graph.

Detailed Description

The class to modify circle models.

Definition at line 33 of file ModelEditor.h.

Constructor & Destructor Documentation

ModelEditor()

ModelEditor::ModelEditor ( std::shared_ptr< CircleModel >  circle_model )

explicit

Initialize the editor with CircleModel object.

Definition at line 67 of file ModelEditor.cpp.

                                                                : _circle_model{circle_model}
{
  assert(circle_model != nullptr); // FIX_CALLER_UNLESS
}

Member Function Documentation

resize_inputs()

ModelEditor & ModelEditor::resize_inputs

(

const std::vector< Shape > &

new_inputs_shapes

)

Resize the model. It means changing shape of the inputs and propagating changes through the graph.

Exceptions:

• std::runtime_error if the new_inputs_shapes are invalid. It can happens for scenarios like:
  • new shapes for NOT all inputs are provided
  • an exception was thrown during shape inference pass

Definition at line 72 of file ModelEditor.cpp.

{
  auto graph = _circle_model->module()->graph();
  change_inputs_shapes(graph, new_inputs_shapes);
 
  logo::Phase phase;
  phase.emplace_back(std::make_unique<logo::RemoveDeadNodeWithQueryPass>());
  phase.emplace_back(std::make_unique<luci::CircleShapeInferencePass>());
  phase.emplace_back(std::make_unique<luci::CircleTypeInferencePass>());
 
  logo::PhaseRunner<logo::PhaseStrategy::Restart> phase_runner{graph};
  try
  {
    phase_runner.run(phase);
  }
  catch (const std::exception &e)
  {
    throw std::runtime_error("Exception during resizing with message: " + std::string{e.what()});
  }
 
  return *this;
}

Referenced by entry().

---

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

• compiler/circle-resizer/include/ModelEditor.h
• compiler/circle-resizer/src/ModelEditor.cpp