MuonML::InferenceUtils Namespace Reference

Classes
struct	SessionBackend

Functions
template<class SessionToolHandle>
SessionBackend	sessionBackend (const SessionToolHandle &sessionTool)
float	sigmoid (float x)

Function Documentation

sessionBackend()

template<class SessionToolHandle>

SessionBackend MuonML::InferenceUtils::sessionBackend

(

const SessionToolHandle &

sessionTool

)

Definition at line 19 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 26 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);
}