Searched refs:getStepCallbacksUnlessEmpty (Results 1 – 7 of 7) sorted by relevance
53 auto step_callbacks = at::getStepCallbacksUnlessEmpty(at::RecordScope::USER_SCOPE); in runPureRecordFunctionBench()
678 auto step_callbacks = at::getStepCallbacksUnlessEmpty(at::RecordScope::FUNCTION);729 auto step_callbacks = at::getStepCallbacksUnlessEmpty(at::RecordScope::FUNCTION);
162 at::getStepCallbacksUnlessEmpty(at::RecordScope::BACKWARD_FUNCTION); in operator()
1336 at::getStepCallbacksUnlessEmpty(at::RecordScope::STATIC_RUNTIME_MODEL); in run_impl_record_functions()1382 at::getStepCallbacksUnlessEmpty(at::RecordScope::STATIC_RUNTIME_MODEL); in run_impl_record_functions_async()2082 at::getStepCallbacksUnlessEmpty(at::RecordScope::STATIC_RUNTIME_OP); in run()
524 TORCH_API std::optional<StepCallbacks> getStepCallbacksUnlessEmpty(
616 std::optional<StepCallbacks> getStepCallbacksUnlessEmpty(RecordScope scope) { in getStepCallbacksUnlessEmpty() function
943 auto step_callbacks = at::getStepCallbacksUnlessEmpty( in checkAndStartRecordFunction()