Home
last modified time | relevance | path

Searched defs:TensorOptionsArguments (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/torchgen/api/
H A Dstructured.py100 def argument(a: Argument | SelfArgument | TensorOptionsArguments) -> list[Binding]:
H A Dnative.py86 a: Argument | SelfArgument | TensorOptionsArguments,
/aosp_15_r20/external/pytorch/tools/autograd/
H A Dgen_trace_type.py142 def dispatch_trace_input(arg: Argument | TensorOptionsArguments) -> Sequence[str]:
H A Dgen_variable_type.py1091 arg: Argument | SelfArgument | TensorOptionsArguments,
/aosp_15_r20/external/pytorch/torchgen/static_runtime/
H A Dgenerator.py30 arguments: Sequence[Argument | SelfArgument | TensorOptionsArguments],
/aosp_15_r20/external/pytorch/torchgen/
H A Dmodel.py2154 class TensorOptionsArguments: class