FlavorTagDiscriminants::OnnxOutput Class Reference

#include <OnnxOutput.h>

Collaboration diagram for FlavorTagDiscriminants::OnnxOutput:

Public Types
enum	OutputType { OutputType::UNKNOWN, OutputType::FLOAT, OutputType::VECCHAR, OutputType::VECFLOAT }

Public Member Functions
	OnnxOutput (const std::string &name, ONNXTensorElementDataType type, int rank)
	OnnxOutput (const std::string &name, ONNXTensorElementDataType type, const std::string &name_in_model)

Public Attributes
const std::string	name
const std::string	name_in_model
const OutputType	type

Private Member Functions
OutputType	getOutputType (ONNXTensorElementDataType type, int rank) const
const std::string	getName (const std::string &name, const std::string &model_name) const

Detailed Description

Definition at line 16 of file OnnxOutput.h.

Member Enumeration Documentation

OutputType

enum FlavorTagDiscriminants::OnnxOutput::OutputType

strong

Enumerator
UNKNOWN
FLOAT
VECCHAR
VECFLOAT

Definition at line 19 of file OnnxOutput.h.

{UNKNOWN, FLOAT, VECCHAR, VECFLOAT};

Constructor & Destructor Documentation

OnnxOutput() [1/2]

FlavorTagDiscriminants::OnnxOutput::OnnxOutput	(	const std::string &	name,
		ONNXTensorElementDataType	type,
		int	rank
	)

Definition at line 12 of file OnnxOutput.cxx.

                       : name(name),
                         name_in_model(name),
                         type(getOutputType(type, rank)){}

OnnxOutput() [2/2]

FlavorTagDiscriminants::OnnxOutput::OnnxOutput	(	const std::string &	name,
		ONNXTensorElementDataType	type,
		const std::string &	name_in_model
	)

Definition at line 20 of file OnnxOutput.cxx.

                       : name(getName(name, model_name)),
                         name_in_model(name),
                         type(getOutputType(type, 0)){}

Member Function Documentation

getName()

const std::string FlavorTagDiscriminants::OnnxOutput::getName ( const std::string &  name, const std::string &  model_name  ) const

private

Definition at line 27 of file OnnxOutput.cxx.

                                                                                            {
  // unfortunately, this is block is needed to support some taggers that we schedule that don't have
  // a well defined model name and rely on output remapping.
  if (model_name == "UnknownModelName") {
    return name;
  }
  return model_name + "_" + name;
}

getOutputType()

OnnxOutput::OutputType FlavorTagDiscriminants::OnnxOutput::getOutputType ( ONNXTensorElementDataType  type, int  rank  ) const

private

Definition at line 36 of file OnnxOutput.cxx.

                                                                                             {
  // Determine the output node type based on the type and shape of the output tensor.
  using ORT = ONNXTensorElementDataType;
  if (type == ORT::ONNX_TENSOR_ELEMENT_DATA_TYPE_FLOAT) {
    if (rank == 0) {
      return OutputType::FLOAT;
    } else if (rank == 1) {
      return OutputType::VECFLOAT;
    }
  } else if (type == ORT::ONNX_TENSOR_ELEMENT_DATA_TYPE_INT8) {
    return OutputType::VECCHAR;
  }
  return OutputType::UNKNOWN;
}

Member Data Documentation

name

const std::string FlavorTagDiscriminants::OnnxOutput::name

Definition at line 31 of file OnnxOutput.h.

name_in_model

const std::string FlavorTagDiscriminants::OnnxOutput::name_in_model

Definition at line 32 of file OnnxOutput.h.

type

const OutputType FlavorTagDiscriminants::OnnxOutput::type

Definition at line 33 of file OnnxOutput.h.

---

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

• OnnxOutput.h
• OnnxOutput.cxx