Home
last modified time | relevance | path

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

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