MuonML::InferenceUtils Namespace Reference

Classes
struct	SessionBackend
struct	SegmentQualityOrder
	Common quality ordering for segment representatives. More...

Functions
template<class SessionToolHandle>
SessionBackend	sessionBackend (const SessionToolHandle &sessionTool)
float	sigmoid (float x)
float	reducedChi2 (const xAOD::MuonSegment &segment)
int	compareFloat (float first, float second)
	Three-way float comparison which orders NaN after all numeric values.
int	compareFloatDescending (float first, float second)
	Three-way descending comparison which also orders NaN last.

Function Documentation

compareFloat()

int MuonML::InferenceUtils::compareFloat ( float first, float second )

inline

Three-way float comparison which orders NaN after all numeric values.

Definition at line 42 of file InferenceUtils.h.

                                                   {
  if (std::isnan(first)) return std::isnan(second) ? 0 : 1;
  if (std::isnan(second)) return -1;
  if (first < second) return -1;
  if (first > second) return 1;
  return 0;
}

compareFloatDescending()

int MuonML::InferenceUtils::compareFloatDescending ( float first, float second )

inline

Three-way descending comparison which also orders NaN last.

Definition at line 51 of file InferenceUtils.h.

                                                             {
  if (std::isnan(first)) return std::isnan(second) ? 0 : 1;
  if (std::isnan(second)) return -1;
  if (first > second) return -1;
  if (first < second) return 1;
  return 0;
}

reducedChi2()

float MuonML::InferenceUtils::reducedChi2 ( const xAOD::MuonSegment & segment )

inline

Definition at line 37 of file InferenceUtils.h.

                                                         {
  return segment.chiSquared() / std::max(1.f, segment.numberDoF());
}

sessionBackend()

template<class SessionToolHandle>

SessionBackend MuonML::InferenceUtils::sessionBackend

(

const SessionToolHandle &

sessionTool

)

Definition at line 21 of file InferenceUtils.h.

                                                                    {
  if (const auto* cudaTool = dynamic_cast<const AthOnnx::OnnxRuntimeSessionToolCUDA*>(sessionTool.get())) {
    return SessionBackend{true, cudaTool->deviceId()};
  }
  return SessionBackend{};
}

sigmoid()

float MuonML::InferenceUtils::sigmoid ( float x )

inline

Definition at line 28 of file InferenceUtils.h.

                              {
  if (x >= 0.f) {
    const float z = std::exp(-x);
    return 1.f / (1.f + z);
  }
  const float z = std::exp(x);
  return z / (1.f + z);
}