Home
last modified time | relevance | path

Searched refs:PythonRecordFunction (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Drecord_function_ops.cpp43 c10::intrusive_ptr<PythonRecordFunction> record_function_enter_new( in record_function_enter_new()
47 c10::make_intrusive<PythonRecordFunction>(at::RecordScope::USER_SCOPE); in record_function_enter_new()
73 const c10::intrusive_ptr<PythonRecordFunction>& record) { in record_function_exit_new()
120 const c10::intrusive_ptr<PythonRecordFunction>& record, in _call_end_callbacks_on_fut_new()
128 m.class_<PythonRecordFunction>("_RecordFunction"); in TORCH_LIBRARY_FRAGMENT()
158 torch::jit::pop(stack).toCustomClass<PythonRecordFunction>(); in TORCH_LIBRARY_FRAGMENT()
H A Drecord_function_ops.h8 struct PythonRecordFunction : public torch::CustomClassHolder { struct
11 explicit PythonRecordFunction( argument
18 TORCH_API c10::intrusive_ptr<PythonRecordFunction> record_function_enter_new( argument
24 const c10::intrusive_ptr<PythonRecordFunction>& record,
H A Dinit.cpp330 using torch::autograd::profiler::PythonRecordFunction; in THPAutograd_initExtension()
331 auto python_rec = c10::make_intrusive<PythonRecordFunction>( in THPAutograd_initExtension()
353 using torch::autograd::profiler::PythonRecordFunction; in THPAutograd_initExtension()
354 auto python_record = torch::jit::toCustomClass<PythonRecordFunction>(obj); in THPAutograd_initExtension()
/aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/
H A Dtorchscript_functions.cpp22 c10::intrusive_ptr<torch::autograd::profiler::PythonRecordFunction> record; in rpcTorchscript()