Searched defs:ProfilerStateBase (Results 1 – 2 of 2) sorted by relevance
119 struct TORCH_API ProfilerStateBase : public c10::MemoryReportingInfoBase { struct124 static ProfilerStateBase* get() { in get() argument132 static std::shared_ptr<ProfilerStateBase> pop() { in pop() argument137 const ProfilerConfig& config() const { in config()144 bool memoryProfilingEnabled() const override { in memoryProfilingEnabled()154 ProfilerConfig config_ = ProfilerConfig(ProfilerState::Disabled);156 at::CallbackHandle handle_ = 0;
97 /*explicit*/ ProfilerStateBase::ProfilerStateBase(ProfilerConfig config) in ProfilerStateBase() function in torch::profiler::impl::ProfilerStateBase