Home
last modified time | relevance | path

Searched defs:step_callbacks (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/binaries/
H A Drecord_function_benchmark.cc53 auto step_callbacks = at::getStepCallbacksUnlessEmpty(at::RecordScope::USER_SCOPE); in runPureRecordFunctionBench() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dimpl.cpp1335 auto step_callbacks = in run_impl_record_functions() local
1381 auto step_callbacks = in run_impl_record_functions_async() local
2081 auto step_callbacks = in run() local
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dfunction.h161 auto step_callbacks = in operator() local
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A Drecord_function.cpp532 RecordFunction::RecordFunction(StepCallbacks&& step_callbacks) in RecordFunction()
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dinterpreter.cpp943 auto step_callbacks = at::getStepCallbacksUnlessEmpty( in checkAndStartRecordFunction() local