#include <OpPrinter.h>

Public Member Functions
	OpPrinterRegistry ()

const OpPrinter *	lookup (tflite::BuiltinOperator op) const

Static Public Member Functions
static OpPrinterRegistry &	get ()

Detailed Description

Definition at line 34 of file OpPrinter.h.

Constructor & Destructor Documentation

◆ OpPrinterRegistry()

tfldump::OpPrinterRegistry::OpPrinterRegistry ( )

Definition at line 741 of file OpPrinter.cpp.

{
  _op_map[tflite::BuiltinOperator_ADD] = make_unique<AddPrinter>();
  // There is no Option for ADD_N
  _op_map[tflite::BuiltinOperator_ARG_MAX] = make_unique<ArgMaxPrinter>();
  _op_map[tflite::BuiltinOperator_ARG_MIN] = make_unique<ArgMinPrinter>();
  _op_map[tflite::BuiltinOperator_AVERAGE_POOL_2D] = make_unique<Pool2DPrinter>();
  _op_map[tflite::BuiltinOperator_BIDIRECTIONAL_SEQUENCE_LSTM] =
    make_unique<BidirectionalSequenceLSTMPrinter>();
  _op_map[tflite::BuiltinOperator_CAST] = make_unique<CastPrinter>();
  // There is no Option for CEIL
  _op_map[tflite::BuiltinOperator_CONCATENATION] = make_unique<ConcatenationPrinter>();
  _op_map[tflite::BuiltinOperator_CONV_2D] = make_unique<Conv2DPrinter>();
  // There is no Option for DENSIFY
  _op_map[tflite::BuiltinOperator_DEPTH_TO_SPACE] = make_unique<DepthToSpacePrinter>();
  _op_map[tflite::BuiltinOperator_DEPTHWISE_CONV_2D] = make_unique<DepthwiseConv2DPrinter>();
  // There is no Option for DEQUANTIZE
  _op_map[tflite::BuiltinOperator_DIV] = make_unique<DivPrinter>();
  _op_map[tflite::BuiltinOperator_FAKE_QUANT] = make_unique<FakeQuantPrinter>();
  // There is no Option for FLOOR
  // There is no Option for FLOOR_MOD
  _op_map[tflite::BuiltinOperator_FULLY_CONNECTED] = make_unique<FullyConnectedPrinter>();
  _op_map[tflite::BuiltinOperator_GATHER] = make_unique<GatherPrinter>();
  _op_map[tflite::BuiltinOperator_GELU] = make_unique<GeluPrinter>();
  _op_map[tflite::BuiltinOperator_IF] = make_unique<IfPrinter>();
  _op_map[tflite::BuiltinOperator_L2_POOL_2D] = make_unique<Pool2DPrinter>();
  _op_map[tflite::BuiltinOperator_L2_NORMALIZATION] = make_unique<L2NormPrinter>();
  _op_map[tflite::BuiltinOperator_LEAKY_RELU] = make_unique<LeakyReluPrinter>();
  _op_map[tflite::BuiltinOperator_LOCAL_RESPONSE_NORMALIZATION] =
    make_unique<LocalResponseNormalizationPrinter>();
  // There is no Option for LOG
  // There is no Option for LOGISTIC
  // There is no Option for LOG_SOFTMAX
  _op_map[tflite::BuiltinOperator_MAX_POOL_2D] = make_unique<Pool2DPrinter>();
  _op_map[tflite::BuiltinOperator_MEAN] = make_unique<ReducerPrinter>();
  _op_map[tflite::BuiltinOperator_MIRROR_PAD] = make_unique<MirrorPadPrinter>();
  _op_map[tflite::BuiltinOperator_MUL] = make_unique<MulPrinter>();
  // There is no Option for NON_MAX_SUPPRESSION_V4
  // There is no Option for NON_MAX_SUPPRESSION_V5
  _op_map[tflite::BuiltinOperator_ONE_HOT] = make_unique<OneHotPrinter>();
  _op_map[tflite::BuiltinOperator_PACK] = make_unique<PackPrinter>();
  // There is no Option for PAD
  // There is no Option for PADV2
  // There is no Option for PRELU
  // There is no Option for RELU
  // There is no Option for RELU6
  // There is no Option for RELU_0_TO_1
  // There is no Option for RELU_N1_TO_1
  _op_map[tflite::BuiltinOperator_REDUCE_ANY] = make_unique<ReducerPrinter>();
  _op_map[tflite::BuiltinOperator_REDUCE_MAX] = make_unique<ReducerPrinter>();
  _op_map[tflite::BuiltinOperator_REDUCE_MIN] = make_unique<ReducerPrinter>();
  _op_map[tflite::BuiltinOperator_REDUCE_PROD] = make_unique<ReducerPrinter>();
  _op_map[tflite::BuiltinOperator_RESHAPE] = make_unique<ReshapePrinter>();
  _op_map[tflite::BuiltinOperator_RESIZE_BILINEAR] = make_unique<ResizeBilinearPrinter>();
  _op_map[tflite::BuiltinOperator_RESIZE_NEAREST_NEIGHBOR] =
    make_unique<ResizeNearestNeighborPrinter>();
  _op_map[tflite::BuiltinOperator_REVERSE_SEQUENCE] = make_unique<ReverseSequencePrinter>();
  // There is no Option for ROUND
  // There is no Option for SELECT
  // There is no Option for SELECT_V2
  _op_map[tflite::BuiltinOperator_SHAPE] = make_unique<ShapePrinter>();
  // There is no Option for SIN
  // There is no Option for SLICE
  _op_map[tflite::BuiltinOperator_SOFTMAX] = make_unique<SoftmaxPrinter>();
  _op_map[tflite::BuiltinOperator_SPACE_TO_DEPTH] = make_unique<SpaceToDepthPrinter>();
  // There is no Option for SPACE_TO_BATCH_ND
  _op_map[tflite::BuiltinOperator_SPARSE_TO_DENSE] = make_unique<SparseToDensePrinter>();
  _op_map[tflite::BuiltinOperator_SPLIT] = make_unique<SplitPrinter>();
  _op_map[tflite::BuiltinOperator_SPLIT_V] = make_unique<SplitVPrinter>();
  _op_map[tflite::BuiltinOperator_SQUEEZE] = make_unique<SqueezePrinter>();
  _op_map[tflite::BuiltinOperator_STRIDED_SLICE] = make_unique<StridedSlicePrinter>();
  _op_map[tflite::BuiltinOperator_SUB] = make_unique<SubPrinter>();
  _op_map[tflite::BuiltinOperator_SUM] = make_unique<ReducerPrinter>();
  _op_map[tflite::BuiltinOperator_SVDF] = make_unique<SVDFPrinter>();
  _op_map[tflite::BuiltinOperator_TRANSPOSE_CONV] = make_unique<TransposeConvPrinter>();
  // There is no Option for TOPK_V2
  _op_map[tflite::BuiltinOperator_UNIDIRECTIONAL_SEQUENCE_LSTM] =
    make_unique<UnidirectionalSequenceLSTMPrinter>();
  _op_map[tflite::BuiltinOperator_UNIQUE] = make_unique<UniquePrinter>();
  _op_map[tflite::BuiltinOperator_WHILE] = make_unique<WhilePrinter>();
  _op_map[tflite::BuiltinOperator_CUSTOM] = make_unique<CustomOpPrinter>();
}

Member Function Documentation

◆ get()

static OpPrinterRegistry & tfldump::OpPrinterRegistry::get ( )

inlinestatic

Definition at line 49 of file OpPrinter.h.

  {
    static OpPrinterRegistry me;
    return me;
  }

Referenced by tfldump::dump_sub_graph().

◆ lookup()

const OpPrinter * tfldump::OpPrinterRegistry::lookup ( tflite::BuiltinOperator op ) const

inline

Definition at line 40 of file OpPrinter.h.

  {
    if (_op_map.find(op) == _op_map.end())
      return nullptr;
 
    return _op_map.at(op).get();
  }

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

compiler/tfldump/src/OpPrinter.h
compiler/tfldump/src/OpPrinter.cpp