Searched defs:step_callbacks (Results 1 – 5 of 5) sorted by relevance
53 auto step_callbacks = at::getStepCallbacksUnlessEmpty(at::RecordScope::USER_SCOPE); in runPureRecordFunctionBench() local
1335 auto step_callbacks = in run_impl_record_functions() local1381 auto step_callbacks = in run_impl_record_functions_async() local2081 auto step_callbacks = in run() local
161 auto step_callbacks = in operator() local
532 RecordFunction::RecordFunction(StepCallbacks&& step_callbacks) in RecordFunction()
943 auto step_callbacks = at::getStepCallbacksUnlessEmpty( in checkAndStartRecordFunction() local