7 #include <unordered_map> 9 #include <ATen/core/interned_strings.h> 10 #include <c10/util/Exception.h> 16 Symbol symbol(
const std::string& s);
17 std::pair<const char*, const char*> string(
Symbol sym);
22 Symbol _symbol(
const std::string& s);
23 std::pair<const char*, const char*> customString(
Symbol sym);
24 std::unordered_map<std::string, Symbol> string_to_sym_;
28 std::string qual_name;
29 std::string unqual_name;
31 std::vector<SymbolInfo> sym_to_info_;
To register your own kernel for an operator, do in one (!) cpp file: C10_REGISTER_KERNEL(OperatorHand...