Home
last modified time | relevance | path

Searched refs:getCopyOfFuncTorchTLS (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DFuncTorchTLS.cpp11 std::unique_ptr<FuncTorchTLSBase> getCopyOfFuncTorchTLS() { in getCopyOfFuncTorchTLS() function
H A DFuncTorchTLS.h37 TORCH_API std::unique_ptr<FuncTorchTLSBase> getCopyOfFuncTorchTLS();
H A DThreadLocalState.cpp17 rf_tls_(at::get_record_function_tls_()), functorch_tls_(functorch::getCopyOfFuncTorchTLS()), in ThreadLocalState()