Searched refs:normalize_function (Results 1 – 15 of 15) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/nested/_internal/ |
| H A D | ops.py | 9 from torch.fx.operator_schemas import normalize_function
327 _, new_kwargs = normalize_function( # type: ignore[misc]
407 _, new_kwargs = normalize_function( # type: ignore[misc]
433 _, new_kwargs = normalize_function( # type: ignore[misc]
462 _, new_kwargs = normalize_function( # type: ignore[misc]
476 _, new_kwargs = normalize_function( # type: ignore[misc]
495 _, new_kwargs = normalize_function( # type: ignore[misc]
508 _, new_kwargs = normalize_function( # type: ignore[misc]
542 _, new_kwargs = normalize_function( # type: ignore[misc]
570 _, new_kwargs = normalize_function( # type: ignore[misc]
[all …]
|
| /aosp_15_r20/external/pytorch/torch/_subclasses/ |
| H A D | fake_impls.py | 29 from torch.fx.operator_schemas import normalize_function
154 _, new_kwargs = normalize_function(
182 _, new_kwargs = normalize_function(
196 _, new_kwargs = normalize_function(
497 _, new_kwargs = normalize_function(
566 _, new_kwargs = normalize_function(
603 _, new_kwargs = normalize_function(
613 _, new_kwargs = normalize_function(
662 _, kwargs = normalize_function(
|
| H A D | schema_check_mode.py | 8 from torch.fx.operator_schemas import normalize_function
142 pre_arguments = normalize_function(
|
| H A D | fake_tensor.py | 51 from torch.fx.operator_schemas import normalize_function
2209 _, new_kwargs = normalize_function( # type: ignore[misc]
|
| /aosp_15_r20/external/pytorch/test/export/ |
| H A D | opinfo_schema.py | 6 from torch.fx.operator_schemas import normalize_function
50 pre_arguments = normalize_function(
|
| /aosp_15_r20/external/pytorch/torch/fx/experimental/ |
| H A D | normalize.py | 12 normalize_function,
75 new_args_and_kwargs = normalize_function(
|
| /aosp_15_r20/external/pytorch/torch/fx/passes/ |
| H A D | graph_drawer.py | 12 from torch.fx.operator_schemas import normalize_function
254 args, kwargs = normalize_function( # type: ignore[misc]
|
| /aosp_15_r20/external/executorch/exir/ |
| H A D | tracer.py | 52 from torch.fx.operator_schemas import normalize_function
276 args, kwargs = normalize_function(func, args, kwargs) # pyre-fixme[23]
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_schema_check.py | 10 from torch.fx.operator_schemas import normalize_function
466 named_arg_list = normalize_function(
|
| H A D | test_fx_experimental.py | 37 normalize_function,
1608 norm_args_and_kwargs = normalize_function(
1697 norm_args_and_kwargs = normalize_function(
1721 …args, kwargs = normalize_function(target, (inp1,), {"the_template": inp2}, normalize_to_only_use_k…
|
| /aosp_15_r20/external/pytorch/torch/fx/ |
| H A D | node.py | 10 from torch.fx.operator_schemas import normalize_function, normalize_module, ArgsKwargsPair
704 …return normalize_function(self.target, self.args, self.kwargs, arg_types, kwarg_types) # type: ig…
|
| H A D | operator_schemas.py | 271 def normalize_function( function
|
| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
| H A D | post_grad.py | 1089 from torch.fx.operator_schemas import normalize_function
1103 args_, kwargs_ = normalize_function(node.target, node.args, node.kwargs) # type: ignore[misc]
|
| /aosp_15_r20/external/pytorch/torch/utils/ |
| H A D | _python_dispatch.py | 623 new_args, new_kwargs = torch.fx.operator_schemas.normalize_function( # type: ignore[misc]
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | pattern_matcher.py | 622 from torch.fx.operator_schemas import normalize_function
624 normalized_args_and_kwargs = normalize_function(
|