[Codemod][Modernize Containers] fbcode//caffe2:torch_sources [A] [A]

Reviewed By: dtolnay

Differential Revision: D78953203

aten/src/ATen/ThreadLocalPythonObjects.cpp

@@ -14,12 +14,12 @@ void ThreadLocalPythonObjects::set(const std::string& key, std::shared_ptr<SafeP
 }
 
 const std::shared_ptr<SafePyObject>& ThreadLocalPythonObjects::get(const std::string& key) {
-  TORCH_CHECK(py_objects.obj_dict_.count(key));
+  TORCH_CHECK(py_objects.obj_dict_.contains(key));
   return py_objects.obj_dict_[key];
 }
 
 bool ThreadLocalPythonObjects::contains(const std::string& key) {
-  return py_objects.obj_dict_.count(key);
+  return py_objects.obj_dict_.contains(key);
 }
 
 void ThreadLocalPythonObjects::set_state(ThreadLocalPythonObjects state) {

aten/src/ATen/core/custom_class.cpp

@@ -71,7 +71,7 @@ void registerCustomClass(at::ClassTypePtr class_type) {
   // NOLINTNEXTLINE(bugprone-unchecked-optional-access)
   auto name = class_type->name()->qualifiedName();
   TORCH_CHECK(
-      !customClasses().count(name),
+      !customClasses().contains(name),
       "Custom class with name ",
       name,
       " is already registered. Ensure that registration with torch::class_ is only called once.");
@@ -80,7 +80,7 @@ void registerCustomClass(at::ClassTypePtr class_type) {
 
 at::ClassTypePtr getCustomClass(const std::string& class_name) {
   auto ret =
-      customClasses().count(class_name) ? customClasses()[class_name] : nullptr;
+      customClasses().contains(class_name) ? customClasses()[class_name] : nullptr;
   if (ret) {
     RECORD_CUSTOM_CLASS(class_name);
   }