Home
last modified time | relevance | path

Searched refs:GetPythonFramesFunction (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Ddebug_util.cpp58 std::function<std::vector<SourceLocation>()>& GetPythonFramesFunction() { in GetPythonFramesFunction() function
70 if (!torch::lazy::GetPythonFramesFunction()) { in GetFirstUserFrameInPython()
74 auto frames = torch::lazy::GetPythonFramesFunction()(); in GetFirstUserFrameInPython()
120 std::vector<SourceLocation> frames = GetPythonFramesFunction()(); in GetTensorsGraphInfo()
H A Ddebug_util.h12 GetPythonFramesFunction();
H A Dir_metadata.cpp102 meta.frame_info = torch::lazy::GetPythonFramesFunction()(); in GetMetaDataIfDebugging()
/aosp_15_r20/external/pytorch/test/cpp/lazy/
H A Dtest_ir.cpp85 GetPythonFramesFunction() = [&]() -> std::vector<SourceLocation> { in TEST()
/aosp_15_r20/external/pytorch/torch/csrc/lazy/python/
H A Dinit.cpp335 GetPythonFramesFunction() = GetPythonFrames; in initLazyBindings()