/aosp_15_r20/external/pytorch/torch/fx/ |
H A D | interpreter.py | 300 …def call_module(self, target : 'Target', args : Tuple[Argument, ...], kwargs : Dict[str, Any]) -> … member in Interpreter 490 …def call_module(self, target : 'Target', args : Tuple[Argument, ...], kwargs : Dict[str, Any]) -> … member in Transformer
|
H A D | _symbolic_trace.py | 480 def call_module( member in Tracer
|
H A D | graph.py | 1220 def call_module(self, member in Graph
|
/aosp_15_r20/external/pytorch/torch/fx/experimental/ |
H A D | schema_type_annotation.py | 61 def call_module(self, target : Target, args : Tuple[Argument, ...], kwargs : Dict[str, Any]): member in AnnotateTypesWithSchema
|
H A D | normalize.py | 91 def call_module( member in NormalizeArgs
|
H A D | optimization.py | 90 …def call_module(self, target : Target, args : Tuple[Argument, ...], kwargs : Dict[str, Any]) -> An… member in remove_dropout.DropoutRemover
|
H A D | meta_tracer.py | 209 def call_module(self, m, forward, args, kwargs): member in MetaTracer
|
H A D | proxy_tensor.py | 1032 def call_module( member in PythonKeyTracer 1678 def call_module( member in _ModuleStackTracer
|
/aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/ |
H A D | fx_onnx_interpreter.py | 697 def call_module( member in FxOnnxInterpreter
|
/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
H A D | pre_grad.py | 347 def call_module(self, target, args, kwargs): member in remove_identity.IdentityRemover
|
/aosp_15_r20/external/executorch/exir/ |
H A D | tracer.py | 411 def call_module( member in DispatchTracer
|
H A D | pass_base.py | 363 def call_module( member in _ExportPassBase.ExportInterpreter
|
/aosp_15_r20/external/pytorch/torch/_export/ |
H A D | pass_base.py | 225 def call_module( member in _ExportPassBaseDeprecatedDoNotUse.ExportInterpreter
|
/aosp_15_r20/external/pytorch/torch/distributed/pipelining/ |
H A D | _IR.py | 371 def call_module(self, target, args, kwargs): member in DetachExecutor
|
/aosp_15_r20/external/pytorch/torch/_inductor/ |
H A D | pattern_matcher.py | 1042 call_module = None # type: ignore[assignment] variable in ReplacementPatternEntry.replace_with_graph.Replacer 1774 call_module = _not_implemented variable in fx_to_pattern.Converter
|
H A D | graph.py | 1024 def call_module(self, target, args, kwargs): member in GraphLowering
|
/aosp_15_r20/external/executorch/exir/emit/ |
H A D | _emitter.py | 1303 def call_module( # pyre-fixme[14] member in _Emitter
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/python/ |
H A D | xla.py | 566 def call_module(args, *, module, Tout, Sout, dim_args_spec=()): function
|
/aosp_15_r20/external/pytorch/test/profiler/ |
H A D | test_profiler.py | 259 def call_module(x): function
|
/aosp_15_r20/external/pytorch/torch/testing/_internal/ |
H A D | common_quantization.py | 104 def call_module(cls, target): member in NodeSpec
|