API

Utils

onnxcustom.utils.measure_time(stmt, context, repeat=10, number=50, div_by_number=False)[source]

Measures a statement and returns the results as a dictionary.

Parameters
  • stmt – string

  • context – variable to know in a dictionary

  • repeat – average over repeat experiment

  • number – number of executions in one row

  • div_by_number – divide by the number of executions

Returns

dictionary

<<<

from onnxcustom.utils import measure_time
from math import cos

res = measure_time("cos(x)", context=dict(cos=cos, x=5.))
print(res)

>>>

    {'average': 1.9085872918367385e-05, 'deviation': 1.3443500584768589e-06, 'min_exec': 1.7879996448755264e-05, 'max_exec': 2.2919848561286926e-05, 'repeat': 10, 'number': 50}

See Timer.repeat for a better understanding of parameter repeat and number. The function returns a duration corresponding to number times the execution of the main statement.