Home
last modified time | relevance | path

Searched defs:FunctionSchema (Results 1 – 25 of 34) sorted by relevance

12

/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A Dfunction_schema.h228 struct TORCH_API FunctionSchema { struct
312 std::vector<Argument> arguments_;
313 std::vector<Argument> returns_;
318 bool is_vararg_;
319 bool is_varret_;
325 std::optional<AliasAnalysisKind> alias_kind_;
330 void checkSchema() const { in checkSchema()
355 const OperatorName& operator_name() const { in operator_name()
358 const std::string& name() const { in name()
361 const std::string& overload_name() const { in overload_name()
[all …]
/aosp_15_r20/external/pytorch/torchgen/
H A Dnative_function_generation.py127 def self_to_out_signature(func: FunctionSchema) -> FunctionSchema:
160 def functional_to_out_signature(func: FunctionSchema) -> FunctionSchema:
182 func: FunctionSchema,
234 def mutable_to_out_signature(func: FunctionSchema) -> FunctionSchema:
514 def gather_nonaliased_inner_rets(func: FunctionSchema, out_var: str) -> list[str]:
H A Dgen_vmap_plumbing.py82 outer_sig: DispatcherSignature, schema: FunctionSchema, cur_level_var: str
127 def accepts_at_least_one_tensor_input(schema: FunctionSchema) -> bool:
H A Dgen_functionalization_type.py185 def wrapper_name(func: FunctionSchema) -> str:
275 def assert_view_op_properties(func: FunctionSchema) -> None:
H A Dgen_aoti_c_shim.py194 def gen_returns(schema: FunctionSchema) -> tuple[list[str], list[str]]:
246 schema: FunctionSchema, device: str, backend_call: str
/aosp_15_r20/external/pytorch/torch/_library/
H A Dutils.py181 schema: _C.FunctionSchema, args: Tuple[Any, ...], kwargs: Dict[str, Any]
285 def has_kwarg_only_args(schema: _C.FunctionSchema):
289 def has_kwarg_only_tensors(schema: _C.FunctionSchema):
299 def has_tensor_arg(schema: _C.FunctionSchema) -> bool:
310 def get_device_arg_index(schema: _C.FunctionSchema) -> Union[int, None]:
/aosp_15_r20/external/pytorch/torchgen/api/
H A Dfunctionalization.py127 def capture_arguments(func: FunctionSchema, *, is_reverse: bool) -> list[Binding]:
145 def returns_type(func: FunctionSchema) -> CType:
162 def inner_call_index(func: FunctionSchema) -> Binding | None:
172 def inner_arguments(func: FunctionSchema, is_reverse: bool) -> list[Binding]:
H A Dautograd.py342 function_schema: FunctionSchema,
368 functional_info_by_signature: dict[
371 non_functional_info_by_signature: dict[
584 differentiability_infos: dict[FunctionSchema, dict[str, DifferentiabilityInfo]],
H A Ddispatcher.py35 def name(func: FunctionSchema) -> str:
81 def jit_arguments(func: FunctionSchema) -> list[Argument]:
119 def arguments(func: FunctionSchema, *, symint: bool = True) -> list[Binding]:
H A Dnative.py47 def name(func: FunctionSchema) -> str:
149 def arguments(func: FunctionSchema, *, symint: bool) -> list[Binding]:
H A Dufunc.py28 def schema_kernel_name(func: FunctionSchema, dispatch_key: DispatchKey) -> str:
H A Dcpp.py73 func: FunctionSchema,
H A Dlazy.py340 func: FunctionSchema,
/aosp_15_r20/external/pytorch/torchgen/static_runtime/
H A Dgenerator.py448 def generate_test_value_definitions(schema: FunctionSchema, index: int) -> str:
462 def generate_test_value_names(schema: FunctionSchema, index: int) -> str:
479 schema: FunctionSchema,
498 def generate_arg_extraction(schema: FunctionSchema) -> str:
593 def should_check_resize(schema: FunctionSchema) -> bool:
/aosp_15_r20/external/executorch/exir/operator/
H A Dconvert.py53 pybind_schema: torch._C.FunctionSchema,
97 def get_out_args_from_schema(out_var_schema: FunctionSchema) -> Tuple[str]:
131 def schema_to_opoverload(schema: FunctionSchema) -> OpOverload:
/aosp_15_r20/external/pytorch/tools/autograd/
H A Dgen_autograd_functions.py450 differentiability_infos: dict[FunctionSchema, dict[str, DifferentiabilityInfo]]
463 differentiability_infos: dict[FunctionSchema, dict[str, DifferentiabilityInfo]],
497 differentiability_infos: dict[FunctionSchema, dict[str, DifferentiabilityInfo]],
H A Dload_derivatives.py68 infos: dict[FunctionSchema, dict[str, DifferentiabilityInfo]],
445 functions_by_signature: dict[FunctionSchema, list[NativeFunction]],
/aosp_15_r20/external/pytorch/torch/_export/passes/
H A Dreplace_view_ops_with_view_copy_ops_pass.py17 def is_view_op(schema: torch._C.FunctionSchema) -> bool:
24 def get_view_copy_of_view_op(schema: torch._C.FunctionSchema) -> Optional[OpOverload]:
/aosp_15_r20/external/executorch/exir/
H A Dcommon.py65 schema: torch._C.FunctionSchema, keyword_args: Dict[str, T]
88 def format_schema_name(schema: torch._C.FunctionSchema) -> str:
/aosp_15_r20/external/executorch/exir/dialects/edge/spec/
H A Dutils.py47 func_schema: torch._C.FunctionSchema,
87 op_name: str, func_schema: torch._C.FunctionSchema
H A Dgen.py115 def get_func_name_yaml(func_schema: torch._C.FunctionSchema) -> str:
/aosp_15_r20/external/pytorch/torch/fx/
H A Doperator_schemas.py68 def _torchscript_schema_to_signature_impl(ts_schema : torch._C.FunctionSchema) -> inspect.Signature:
104 def _torchscript_schema_to_signature(ts_schema : torch._C.FunctionSchema) -> inspect.Signature:
/aosp_15_r20/external/pytorch/torch/_custom_op/
H A Dimpl.py462 def validate_schema(schema: FunctionSchema) -> None:
508 schema: FunctionSchema, func: typing.Callable
/aosp_15_r20/external/pytorch/torch/onnx/
H A D_onnx_supported_ops.py11 def __init__(self, schema: Union[_C.FunctionSchema, str]) -> None:
/aosp_15_r20/external/pytorch/torch/cuda/
H A D_sanitizer.py467 schema: torch.FunctionSchema, args: Tuple[Any, ...], kwargs: Dict[str, Any]
507 schema: torch.FunctionSchema,

12