Home
last modified time | relevance | path

Searched defs:_TensorOrTensors (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/autograd/
H A D__init__.py233 tensors: Optional[_TensorOrTensors], length: int
243 tensors: _TensorOrTensors,
244 grad_tensors: Optional[_TensorOrTensors] = None,
247 grad_variables: Optional[_TensorOrTensors] = None,
361 grad_outputs: Optional[_TensorOrTensors] = None,
H A Dgradcheck.py1951 func: Callable[..., Union[_TensorOrTensors]], # See Note [VarArg of Tensors]
1952 inputs: _TensorOrTensors,
2116 func: Callable[..., _TensorOrTensors], # See Note [VarArg of Tensors] argument
2117 inputs: _TensorOrTensors,
2118 grad_outputs: Optional[_TensorOrTensors] = None,
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_nn.py3216 input: _TensorOrTensors, output: torch.Tensor,
3247 …def _analytical_jacobian(self, module, input: _TensorOrTensors, jacobian_input=True, jacobian_para…
3289 …def _numerical_jacobian(self, module, input: _TensorOrTensors, jacobian_input=True, jacobian_param…
3306 def check_jacobian(self, module, input: _TensorOrTensors, jacobian_input=True):
/aosp_15_r20/external/pytorch/torch/
H A Dtypes.py41 _TensorOrTensors: TypeAlias = Union[Tensor, Sequence[Tensor]] # noqa: PYI047 variable
/aosp_15_r20/external/pytorch/test/
H A Dtest_nn.py79 def _forward(self, module, input: _TensorOrTensors):
86 def _backward(self, module, input: _TensorOrTensors, output, grad_output, create_graph=False):