| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | operator_name.h | 16 std::string overload_name; member
59 c10::string_view overload_name; member
|
| H A D | function_schema.h | 361 const std::string& overload_name() const { in overload_name() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/dispatch/ |
| H A D | Dispatcher.cpp | 85 std::optional<OperatorHandle> Dispatcher::findOp(const OperatorName& overload_name) { in findOp()
130 std::optional<OperatorHandle> Dispatcher::findSchema(const OperatorName& overload_name) { in findSchema()
143 OperatorHandle Dispatcher::findSchemaOrThrow(const char* name, const char* overload_name) { in findSchemaOrThrow()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/python/ |
| H A D | pybind_utils.cpp | 823 const std::string& overload_name, in _maybe_handle_torch_function()
886 std::string overload_name = operations[0]->schema().overload_name(); in _get_operation_for_overload_or_packet() local
|
| H A D | init.cpp | 1648 [](const std::string& op_name, const std::string& overload_name) { in initJITBindings()
1669 const std::string& overload_name) -> std::optional<py::tuple> { in initJITBindings()
|
| H A D | script_init.cpp | 338 auto overload_name = overload_params[i].ident().name(); in mergeDefaultsAndExtraParametersToOverloadDecl() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/ |
| H A D | upgrader_mobile.h | 27 const std::string overload_name; member
|
| H A D | function.cpp | 52 const std::string& overload_name, in append_operator()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/op_registration/ |
| H A D | infer_schema.cpp | 28 std::string&& overload_name, in make_function_schema()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | operator.h | 331 const std::string overload_name, in OperatorGenerator()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/frontend/ |
| H A D | function_schema_parser.cpp | 113 std::string overload_name = ""; in parseName() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_eager/ |
| H A D | kernel_holder.cpp | 469 const auto& overload_name = in produce_aoti_kernel_lib() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | qlinear_prepack.cpp | 340 const char* overload_name, in callOpByName()
|
| /aosp_15_r20/external/pytorch/torch/csrc/ |
| H A D | PyInterpreter.cpp | 962 const auto& overload_name = schema.overload_name(); in getTorchApiFunction() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/utils/ |
| H A D | python_dispatch.cpp | 532 [](const char* name, const char* overload_name) -> c10::OperatorHandle { in initDispatchBindings()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | vulkan_quantized_api_test.cpp | 137 const char* overload_name, in callOpByName()
|
| H A D | vulkan_api_test.cpp | 250 const char* overload_name, in callOpByName()
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_utils.py | 4128 def assertAtenOp(self, onnx_model, operator, overload_name=""): argument
|