Home
last modified time | relevance | path

Searched defs:call_module (Results 1 – 20 of 20) sorted by relevance

/aosp_15_r20/external/pytorch/torch/fx/
H A Dinterpreter.py300 …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.py480 def call_module( member in Tracer
H A Dgraph.py1220 def call_module(self, member in Graph
/aosp_15_r20/external/pytorch/torch/fx/experimental/
H A Dschema_type_annotation.py61 def call_module(self, target : Target, args : Tuple[Argument, ...], kwargs : Dict[str, Any]): member in AnnotateTypesWithSchema
H A Dnormalize.py91 def call_module( member in NormalizeArgs
H A Doptimization.py90 …def call_module(self, target : Target, args : Tuple[Argument, ...], kwargs : Dict[str, Any]) -> An… member in remove_dropout.DropoutRemover
H A Dmeta_tracer.py209 def call_module(self, m, forward, args, kwargs): member in MetaTracer
H A Dproxy_tensor.py1032 def call_module( member in PythonKeyTracer
1678 def call_module( member in _ModuleStackTracer
/aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/
H A Dfx_onnx_interpreter.py697 def call_module( member in FxOnnxInterpreter
/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/
H A Dpre_grad.py347 def call_module(self, target, args, kwargs): member in remove_identity.IdentityRemover
/aosp_15_r20/external/executorch/exir/
H A Dtracer.py411 def call_module( member in DispatchTracer
H A Dpass_base.py363 def call_module( member in _ExportPassBase.ExportInterpreter
/aosp_15_r20/external/pytorch/torch/_export/
H A Dpass_base.py225 def call_module( member in _ExportPassBaseDeprecatedDoNotUse.ExportInterpreter
/aosp_15_r20/external/pytorch/torch/distributed/pipelining/
H A D_IR.py371 def call_module(self, target, args, kwargs): member in DetachExecutor
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dpattern_matcher.py1042 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 Dgraph.py1024 def call_module(self, target, args, kwargs): member in GraphLowering
/aosp_15_r20/external/executorch/exir/emit/
H A D_emitter.py1303 def call_module( # pyre-fixme[14] member in _Emitter
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/python/
H A Dxla.py566 def call_module(args, *, module, Tout, Sout, dim_args_spec=()): function
/aosp_15_r20/external/pytorch/test/profiler/
H A Dtest_profiler.py259 def call_module(x): function
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_quantization.py104 def call_module(cls, target): member in NodeSpec