Home
last modified time | relevance | path

Searched refs:OperatorName (Results 1 – 25 of 68) sorted by relevance

123

/aosp_15_r20/external/pytorch/aten/src/ATen/core/dispatch/
H A DDispatcher.cpp85 std::optional<OperatorHandle> Dispatcher::findOp(const OperatorName& overload_name) { in findOp()
86 …return operatorLookupTable_.read([&] (const ska::flat_hash_map<OperatorName, OperatorHandle>& oper… in findOp() argument
112 void Dispatcher::waitForImpl(const OperatorName& op_name, std::optional<c10::DispatchKey> maybe_dk)… in waitForImpl()
130 std::optional<OperatorHandle> Dispatcher::findSchema(const OperatorName& overload_name) { in findSchema()
159 const std::vector<OperatorName> Dispatcher::getAllOpNames() { in getAllOpNames()
160 …_.read([&] (const ska::flat_hash_map<OperatorName, OperatorHandle>& operatorLookupTable) -> std::v… in getAllOpNames() argument
161 std::vector<OperatorName> allOpNames; in getAllOpNames()
171 OperatorHandle Dispatcher::findOrRegisterName_(const OperatorName& op_name) { in findOrRegisterName_()
177 operators_.emplace_back(OperatorName(op_name)); in findOrRegisterName_()
179 …operatorLookupTable_.write([&] (ska::flat_hash_map<OperatorName, OperatorHandle>& operatorLookupTa… in findOrRegisterName_() argument
[all …]
H A DDispatcher.h72 explicit OperatorDef(OperatorName&& op_name) in OperatorDef()
140 std::optional<OperatorHandle> findSchema(const OperatorName& operator_name);
158 std::optional<OperatorHandle> findOp(const OperatorName& operator_name);
161 const std::vector<OperatorName> getAllOpNames();
199 void waitForImpl(const OperatorName& op_name, std::optional<DispatchKey> dispatch_key);
224 …RegistrationHandleRAII registerImpl(OperatorName op_name, std::optional<DispatchKey> dispatch_key,…
230 …RegistrationHandleRAII registerPythonModule(const OperatorName& op_name, const char* pymodule, con…
235 void throwIfHasPythonModule(OperatorName op_name);
237 std::optional<std::pair<const char*, const char*>> getPyStub(OperatorName op_name);
242 RegistrationHandleRAII registerName(OperatorName op_name);
[all …]
H A DOperatorEntry.h72 explicit OperatorEntry(OperatorName&& operator_name);
106 const OperatorName& operator_name() const { in operator_name()
223 OperatorName name_;
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A Doperator_name.h14 struct OperatorName final {
17 OperatorName(std::string name, std::string overload_name) in OperatorName() function
75 inline bool operator==(const OperatorName& lhs, const OperatorName& rhs) {
79 inline bool operator!=(const OperatorName& lhs, const OperatorName& rhs) {
83 TORCH_API std::string toString(const OperatorName& opName);
84 TORCH_API std::ostream& operator<<(std::ostream&, const OperatorName&);
90 struct hash<::c10::OperatorName> {
91 size_t operator()(const ::c10::OperatorName& x) const {
H A Dlibrary.cpp100 Library& Library::_def(c10::FunctionSchema&& schema, c10::OperatorName* out_name, const std::vector… in _def()
163 Library& Library::_def(std::variant<c10::OperatorName, c10::FunctionSchema>&& name_or_schema, CppFu… in _def() argument
169 c10::OperatorName name = std::get<c10::OperatorName>(std::move(name_or_schema)); in _def()
181 c10::OperatorName name("", ""); // Get the namespaced name for the impl call in _def()
200 at::OperatorName Library::_parseNameForLib(const char* name_str) const { in _parseNameForLib()
226 at::OperatorName name = _parseNameForLib(name_str); in _impl()
259 c10::OperatorName Library::_resolve(const char* name_str) const { in _resolve()
H A DATenOpList.h6 struct OperatorName;
12 TORCH_API bool is_custom_op(const c10::OperatorName& opName);
H A Doperator_name.cpp5 std::string toString(const OperatorName& opName) { in toString()
11 std::ostream& operator<<(std::ostream& os, const OperatorName& opName) { in operator <<()
/aosp_15_r20/external/pytorch/torch/csrc/jit/frontend/
H A Dfunction_schema_parser.cpp19 using c10::OperatorName;
34 std::variant<OperatorName, FunctionSchema> parseDeclaration() { in parseDeclaration()
35 OperatorName name = parseName(); in parseDeclaration()
40 return OperatorName(std::move(name)); in parseDeclaration()
107 c10::OperatorName parseName() { in parseName()
131 std::vector<std::variant<OperatorName, FunctionSchema>> parseDeclarations() { in parseDeclarations()
132 std::vector<std::variant<OperatorName, FunctionSchema>> results; in parseDeclarations()
140 std::variant<OperatorName, FunctionSchema> parseExactlyOneDeclaration() { in parseExactlyOneDeclaration()
404 std::variant<OperatorName, FunctionSchema> parseSchemaOrName( in parseSchemaOrName()
424 OperatorName parseName(const std::string& name) { in parseName()
[all …]
H A Dfunction_schema_parser.h15 TORCH_API std::variant<c10::OperatorName, c10::FunctionSchema> parseSchemaOrName(
21 TORCH_API c10::OperatorName parseName(const std::string& name);
/aosp_15_r20/external/pytorch/torchgen/executorch/
H A Dmodel.py17 OperatorName,
141 index: dict[OperatorName, dict[ETKernelKey, BackendMetadata]]
162 kernel_index: dict[OperatorName, dict[ETKernelKey, BackendMetadata]], argument
163 backend_indices: dict[DispatchKey, dict[OperatorName, BackendMetadata]],
175 backend_indices: dict[DispatchKey, dict[OperatorName, BackendMetadata]] argument
178 OperatorName, dict[ETKernelKey, BackendMetadata]
184 self, backend_indices: dict[DispatchKey, dict[OperatorName, BackendMetadata]] argument
193 index: dict[OperatorName, BackendMetadata] = {}
H A Dparse.py15 OperatorName,
81 indices: dict[OperatorName, dict[ETKernelKey, BackendMetadata]] = {}
100 def extract_kernel_fields(es: object) -> dict[OperatorName, dict[str, Any]]:
104 fields: dict[OperatorName, dict[str, Any]] = defaultdict(dict)
125 ) -> tuple[list[NativeFunction], dict[OperatorName, dict[str, Any]]]:
/aosp_15_r20/external/pytorch/torchgen/
H A Dgen_lazy_tensor.py22 from torchgen.model import NativeFunction, NativeFunctionsGroup, OperatorName
97 ) -> tuple[list[OperatorName], list[Any], list[OperatorName]]:
108 full_codegen_opnames = [OperatorName.parse(name) for name in full_codegen]
109 ir_gen_opnames = [OperatorName.parse(name) for name in ir_gen]
372 ops_list: list[OperatorName] = full_codegen, argument
H A Dmodel.py528 autogen: list[OperatorName]
545 structured_delegate: OperatorName | None
596 ) -> tuple[NativeFunction, dict[DispatchKey, dict[OperatorName, BackendMetadata]]]:
667 structured_delegate: OperatorName | None = None
669 structured_delegate = OperatorName.parse(structured_delegate_s)
839 else [OperatorName.parse(x) for x in autogen_str.split(", ")]
1280 index: dict[OperatorName, BackendMetadata]
1284 parent_index: dict[DispatchKey, dict[OperatorName, BackendMetadata]], argument
1285 child_index: dict[DispatchKey, dict[OperatorName, BackendMetadata]],
1383 name: OperatorName
[all …]
H A Dgen_aoti_c_shim.py20 OperatorName,
316 func_group_mapping: dict[OperatorName, NativeFunctionsGroup], argument
347 func_group_mapping: dict[OperatorName, NativeFunctionsGroup], argument
371 func_group_mapping: dict[OperatorName, NativeFunctionsGroup], argument
405 func_group_mapping: dict[OperatorName, NativeFunctionsGroup]
427 func_group_mapping: dict[OperatorName, NativeFunctionsGroup], argument
H A Dgen_backend_stubs.py24 OperatorName,
44 native_functions_map: dict[OperatorName, NativeFunction] = {
131 metadata: dict[OperatorName, BackendMetadata] = {}
133 op_name = OperatorName.parse(op)
255 full_codegen: list[OperatorName] | None = None,
273 expected_backend_op_names: dict[OperatorName, str] = dict(
H A Dnative_function_generation.py24 OperatorName,
271 ) -> tuple[NativeFunction, dict[DispatchKey, dict[OperatorName, BackendMetadata]]]:
284 OperatorName(
381 indices: dict[DispatchKey, dict[OperatorName, BackendMetadata]], argument
H A Dgen_executorch.py47 OperatorName,
656 op_to_scoped_name: dict[OperatorName, str] = {
705 dict[DispatchKey, dict[OperatorName, BackendMetadata]] | ETKernelIndex,
742 m: dict[OperatorName, BackendMetadata] argument
743 ) -> dict[OperatorName, BackendMetadata]:
H A Dgen_schema_utils.py13 OperatorName,
96 op_name = OperatorName(BaseOperatorName(op_name, False, False, False), "")
/aosp_15_r20/external/libcxxabi/src/demangle/
H A DItaniumDemangle.h1811 StringView OperatorName; variable
1817 : Node(KFoldExpr), Pack(Pack_), Init(Init_), OperatorName(OperatorName_), in FoldExpr()
1821 F(IsLeftFold, OperatorName, Pack, Init); in match()
1838 S += OperatorName; in printLeft()
1843 S += OperatorName; in printLeft()
1850 S += OperatorName; in printLeft()
1855 S += OperatorName; in printLeft()
4082 StringView OperatorName; in parseFoldExpr() local
4083 if (consumeIf("aa")) OperatorName = "&&"; in parseFoldExpr()
4084 else if (consumeIf("an")) OperatorName = "&"; in parseFoldExpr()
[all …]
/aosp_15_r20/external/swiftshader/third_party/llvm-10.0/llvm/include/llvm/Demangle/
H A DItaniumDemangle.h1915 StringView OperatorName; variable
1921 : Node(KFoldExpr), Pack(Pack_), Init(Init_), OperatorName(OperatorName_), in FoldExpr()
1925 F(IsLeftFold, OperatorName, Pack, Init); in match()
1942 S += OperatorName; in printLeft()
1947 S += OperatorName; in printLeft()
1954 S += OperatorName; in printLeft()
1959 S += OperatorName; in printLeft()
4347 StringView OperatorName; in parseFoldExpr() local
4348 if (consumeIf("aa")) OperatorName = "&&"; in parseFoldExpr()
4349 else if (consumeIf("an")) OperatorName = "&"; in parseFoldExpr()
[all …]
/aosp_15_r20/external/pytorch/aten/src/ATen/core/op_registration/
H A Dop_registration.cpp38 OperatorName name = in checkSchemaAndRegisterOp_()
39 std::get<OperatorName>(std::move(*options.schemaOrName_)); in checkSchemaAndRegisterOp_()
66 const OperatorName& opName, in inferSchemaFromKernels_()
125 OperatorName op_name = schema.operator_name(); in registerOp_()
/aosp_15_r20/external/pytorch/torch/
H A Dlibrary.h434 inline std::variant<c10::OperatorName, c10::FunctionSchema> constructSchemaOrName( in constructSchemaOrName()
438 inline std::variant<c10::OperatorName, c10::FunctionSchema> constructSchemaOrName( in constructSchemaOrName()
439 c10::OperatorName&& n) { in constructSchemaOrName()
442 inline std::variant<c10::OperatorName, c10::FunctionSchema>
716 c10::OperatorName _resolve(const char* name) const;
867 c10::OperatorName* out_name = nullptr,
871 std::variant<c10::OperatorName, c10::FunctionSchema>&&,
880 at::OperatorName _parseNameForLib(const char* name_str) const;
/aosp_15_r20/external/pytorch/tools/test/
H A Dtest_codegen.py27 OperatorName,
335 backend_indices: dict[DispatchKey, dict[OperatorName, BackendMetadata]] = {
387 DispatchKey, dict[OperatorName, BackendMetadata]
446 DispatchKey, dict[OperatorName, BackendMetadata]
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Doperator_name_test.cpp6 c10::OperatorName testName("operator", "operator.overload"); in TEST()
16 c10::OperatorName namespacedName("already_namespaced::operator", "operator.overload"); in TEST()
/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/
H A Dfunction.h78 const c10::OperatorName& opname,
81 TORCH_API std::string operator_str(const c10::OperatorName& opname);

123