xref: /aosp_15_r20/external/pytorch/aten/src/ATen/core/dispatch/ObservedOperators.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1 #pragma once
2 
3 #include <ATen/core/operator_name.h>
4 #include <string>
5 #include <unordered_set>
6 
7 namespace c10 {
8 
9 struct TORCH_API ObservedOperators {
10   ObservedOperators() = delete;
11 
12   static bool isObserved(const OperatorName& name);
13 
14   static std::unordered_set<std::string>& getUnobservedOperatorList();
15 };
16 
17 } // namespace c10
18