Searched defs:OperatorRegistry (Results 1 – 2 of 2) sorted by relevance
17 struct OperatorRegistry { struct19 std::mutex lock;20 OperatorMap operators;23 std::vector<std::shared_ptr<Operator>> to_register;34 std::unordered_map<std::string, std::shared_ptr<Operator>> operators_by_sig;36 operators_by_sig_literal;43 std::unordered_set<c10::OperatorName> registered_operator_names;47 void registerPendingOperators() { in registerPendingOperators()57 void registerOperator(Operator&& op) { in registerOperator()70 void deregisterOperator(const FunctionSchema& schema) { in deregisterOperator()[all …]
128 class OperatorRegistry: class