Home
last modified time | relevance | path

Searched defs:ExecutionTraceObserver (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/profiler/standalone/
H A Dexecution_trace_observer.cpp107 struct TORCH_API ExecutionTraceObserver { // NOLINT struct
108 using ID = size_t;
111 std::map<size_t, std::stack<ID>> opStack{};
114 std::map<const void*, ID> objectId{};
116 enum class RunState { uninitialized, disabled, enabled };
119 std::recursive_mutex gMutex{};
121 std::ofstream out{};
124 std::string fileName{};
127 CallbackHandle cbHandle{INVALID_CALLBACK_HANDLE};
130 int32_t pid{-1};
[all …]
/aosp_15_r20/external/pytorch/torch/profiler/
H A Dprofiler.py801 class ExecutionTraceObserver(_ITraceObserver): class