nnc::DataFormatSwitcher Class Reference

#include <DataFormatSwitcher.h>

Collaboration diagram for nnc::DataFormatSwitcher:

Public Member Functions
	DataFormatSwitcher (mir::DataFormat target_format)
PassData	run (PassData data) override
	run compiler pass
void	cleanup () override
	clean compiler pass data
	~DataFormatSwitcher () override
std::string	getName () override
Public Member Functions inherited from nnc::Pass
virtual	~Pass ()=default

Detailed Description

Definition at line 29 of file DataFormatSwitcher.h.

Constructor & Destructor Documentation

DataFormatSwitcher()

nnc::DataFormatSwitcher::DataFormatSwitcher ( mir::DataFormat  target_format )

explicit

Definition at line 29 of file DataFormatSwitcher.cpp.

  : _target_format(target_format)
{
}

~DataFormatSwitcher()

nnc::DataFormatSwitcher::~DataFormatSwitcher ( )

overridedefault

Member Function Documentation

cleanup()

void nnc::DataFormatSwitcher::cleanup ( )

overridevirtual

clean compiler pass data

Reimplemented from nnc::Pass.

Definition at line 85 of file DataFormatSwitcher.cpp.

{ _candidates_for_switch.clear(); }

getName()

std::string nnc::DataFormatSwitcher::getName ( )

inlineoverridevirtual

Reimplemented from nnc::Pass.

Definition at line 40 of file DataFormatSwitcher.h.

{ return "DataFormatSwitcher"; }

run()

PassData nnc::DataFormatSwitcher::run ( PassData  data )

overridevirtual

run compiler pass

Parameters

data	- data that pass is taken

Returns

data that can be passed to the next pass

Exceptions

PassException

object if errors occured

Implements nnc::Pass.

Definition at line 36 of file DataFormatSwitcher.cpp.

{
  _graph = static_cast<mir::Graph *>(data);
  assert(_graph);
 
  // Collect nodes which use DataFormat
  for (auto *node : _graph->getNodes())
  {
    switch (node->getType())
    { // nodes using DataFormat
      case mir::Operation::Type::avgPool2D:
      case mir::Operation::Type::conv2D:
      case mir::Operation::Type::deConv2D:
      case mir::Operation::Type::depthwiseConv:
      case mir::Operation::Type::maxPool2D:
        _candidates_for_switch.push_back(node);
        break;
      default:
        break; // not use DataFormat
    }
  }
  // Switch collected ops
  for (auto *op : _candidates_for_switch)
  {
    switch (op->getType())
    {
      case mir::Operation::Type::avgPool2D:
        switchAvgPool2D(dynamic_cast<mir::ops::AvgPool2DOp *>(op));
        break;
      case mir::Operation::Type::conv2D:
        switchConv2D(dynamic_cast<mir::ops::Conv2DOp *>(op));
        break;
      case mir::Operation::Type::deConv2D:
        switchDeConv2D(dynamic_cast<mir::ops::DeConv2DOp *>(op));
        break;
      case mir::Operation::Type::depthwiseConv:
        switchDepthwiseConv2D(dynamic_cast<mir::ops::DepthwiseConv2DOp *>(op));
        break;
      case mir::Operation::Type::maxPool2D:
        switchMaxPool2D(dynamic_cast<mir::ops::MaxPool2DOp *>(op));
        break;
      default:
        assert(false && "Can't switch DataFormat for this operation!");
    }
  }
 
  return _graph;
}

References mir::Graph::getNodes().

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

---

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

• compiler/nnc/include/passes/transformations/DataFormatSwitcher.h
• compiler/nnc/passes/transformations/DataFormatSwitcher.cpp