| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | calculate_necessary_args.h | 13 const std::vector<Argument>& schema_args, in CalculateNecessaryArgs() argument
16 if (schema_args.empty()) { in CalculateNecessaryArgs()
21 int64_t schema_idx = static_cast<int64_t>(schema_args.size()) - 1; in CalculateNecessaryArgs()
25 const auto& current_arg = schema_args.at(schema_idx); in CalculateNecessaryArgs()
33 int64_t num_out = static_cast<int64_t>(schema_args.size()) - schema_idx - 1; in CalculateNecessaryArgs()
35 if (schema_args.size() < actual_inputs.size()) { in CalculateNecessaryArgs()
41 schema_idx = schema_args.size() - 1; in CalculateNecessaryArgs()
46 if (!schema_args.at(schema_idx).default_value().has_value()) { in CalculateNecessaryArgs()
50 schema_args.at(schema_idx).default_value().value().toIValue(); in CalculateNecessaryArgs()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | CPUFallback.cpp | 95 auto& schema_args = op.schema().arguments(); in cpu_fallback() local
96 const auto num_arguments = schema_args.size(); in cpu_fallback()
157 const AliasInfo* alias_info = schema_args[tensor_idx].alias_info(); in cpu_fallback()
168 const AliasInfo* alias_info = schema_args[tensorlist_idx].alias_info(); in cpu_fallback()
182 const AliasInfo* alias_info = schema_args[tensorlist_idx].alias_info(); in cpu_fallback()
235 schema_args[input_tensor_idx].alias_info(); in cpu_fallback()
257 schema_args[input_tensor_idx].alias_info(); in cpu_fallback()
|
| /aosp_15_r20/external/pytorch/torch/onnx/_internal/exporter/ |
| H A D | _dispatching.py | 235 schema_args: dict[str, torch.Argument] = {
250 if param.name not in schema_args and param.required:
261 param.name in schema_args
262 and schema_args[param.name].has_default_value()
265 arg = schema_args[param.name].default_value
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/ |
| H A D | ts_eager_fallback.cpp | 211 auto& schema_args = op.schema().arguments(); in ts_eager_fallback() local
212 const auto num_arguments = schema_args.size(); in ts_eager_fallback()
262 const auto alias_info = schema_args[tensor_idx].alias_info(); in ts_eager_fallback()
306 schema_args[input_tensor_idx].alias_info(); in ts_eager_fallback()
|
| /aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_torch/ |
| H A D | oss_proxy_executor.cpp | 229 const std::vector<c10::Argument>& schema_args, in get_input_info_from_serialized() argument
235 auto& schema_arg = schema_args[index]; in get_input_info_from_serialized()
346 const auto& schema_args = schema.arguments(); in OSSProxyExecutor() local
350 get_input_info_from_serialized(schema_args, serialized_node, op_kernel); in OSSProxyExecutor()
|
| H A D | oss_proxy_executor.h | 90 const std::vector<c10::Argument>& schema_args,
|
| /aosp_15_r20/external/pytorch/torch/_custom_op/ |
| H A D | impl.py | 559 def compare(sig_args, schema_args): argument
560 if len(sig_args) != len(schema_args):
562 for (name, param), arg in zip(sig_args, schema_args):
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
| H A D | impl.cpp | 1017 const auto& schema_args = schema->arguments(); in set_inputs() local
1019 DCHECK(!schema_args.empty()); in set_inputs()
1021 args.size() < schema_args.size(), in set_inputs()
1025 std::to_string(schema_args.size() - 1), in set_inputs()
1029 for (const auto i_arg : c10::irange(1, schema_args.size())) { in set_inputs()
1031 const auto& schema_arg = schema_args[i_arg]; in set_inputs()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | ivalue.cpp | 675 const auto& schema_args = lt_schema.arguments(); in checkObjectSortSchema() local
677 (schema_args.size() != 2 || in checkObjectSortSchema()
678 !simpleClassTypeArg(schema_args[0], t) || in checkObjectSortSchema()
679 !simpleClassTypeArg(schema_args[1], t) || in checkObjectSortSchema()
|
| /aosp_15_r20/external/pytorch/torch/cuda/ |
| H A D | _sanitizer.py | 469 schema_args = schema.arguments[: len(args)]
472 yield from zip(schema_args, args)
|
| /aosp_15_r20/external/executorch/exir/serde/ |
| H A D | export_serialize.py | 1898 schema_args = target._schema.arguments
1905 for schema_arg in schema_args:
|
| /aosp_15_r20/external/pytorch/torch/_export/serde/ |
| H A D | serialize.py | 1943 schema_args = _get_schema_from_target(target).arguments
1950 for schema_arg in schema_args:
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | ir.py | 5304 schema_args = schema.arguments
|