Caffe2 - C++ API
A deep learning, cross platform ML framework
thread_name.h
1 #pragma once
2 
3 #include <string>
4 
5 #include "c10/macros/Export.h"
6 
7 namespace c10 {
8 
9 C10_API void setThreadName(std::string name);
10 
11 } // namespace c10
To register your own kernel for an operator, do in one (!) cpp file: C10_REGISTER_KERNEL(OperatorHand...
Definition: alias_info.h:7