Home
last modified time | relevance | path

Searched refs:_as_tuple (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/pytorch/torch/autograd/
H A Dfunctional.py24 def _as_tuple(inp, arg_name=None, fn_name=None): function
324 is_inputs_tuple, inputs = _as_tuple(inputs, "inputs", "vjp")
328 is_outputs_tuple, outputs = _as_tuple(
334 _, v = _as_tuple(v, "v", "vjp")
421 is_inputs_tuple, inputs = _as_tuple(inputs, "inputs", "jvp")
425 _, v = _as_tuple(v, "v", "jvp")
437 is_outputs_tuple, outputs = _as_tuple(
515 is_inputs_tuple, inputs = _as_tuple(inputs, "inputs", "jacobian")
532 _is_outputs_tuple, dual_outputs = _as_tuple(
674 is_inputs_tuple, inputs = _as_tuple(inputs, "inputs", "jacobian")
[all …]
H A Dgradcheck.py284 outputs = _as_tuple(fn(*_as_tuple(inputs)))
552 raw_outputs = _as_tuple(fn(*dual_inputs))
577 raw_outputs = _as_tuple(fn(*dual_inputs))
622 for i, a in enumerate(_as_tuple(inputs))
624 return tuple(a.clone() for a in _as_tuple(fn(*inp)))
683 func_out = _as_tuple(func_out)
1075 dual_outputs = _as_tuple(func(*inputs_with_dual))
1262 raw_outputs = _as_tuple(func(*dual_inputs))
1267 raw_outputs = _as_tuple(func(*dual_inputs))
1359 def _as_tuple(x): function
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/feature_column/
H A Dutils.py115 return _as_tuple(default_value)
117 return _as_tuple(default_value)
130 def _as_tuple(value): function
133 return tuple([_as_tuple(v) for v in value])
/aosp_15_r20/external/python/cpython3/Tools/c-analyzer/c_parser/preprocessor/
Derrors.py7 def _as_tuple(items): function
61 self.argv = _as_tuple(argv) or None
91 self.missing = _as_tuple(missing) or None
/aosp_15_r20/external/pytorch/torch/
H A D_vmap_internals.py41 def _as_tuple( function
125 out_dims_as_tuple = _as_tuple(
/aosp_15_r20/external/fonttools/Lib/fontTools/colorLib/
H A Dbuilder.py487 def _as_tuple(self, paints: Sequence[ot.Paint]) -> Tuple[Any, ...]: member in LayerReuseCache
502 self._as_tuple(layers[lbound:ubound]), -1
517 self.reusePool[self._as_tuple(layers[lbound:ubound])] = (
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_ops.py36 from torch._functorch.eager_transforms import _as_tuple, jvp
205 return _autograd_grad(_as_tuple(result), primals, _as_tuple(cotangents))
269 cotangents = _as_tuple(
292 tangents = _as_tuple(tree_map(lambda x: torch.randn_like(x), primals))
315 tangents = _as_tuple(tree_map(lambda x: torch.randn_like(x), primals))
323 tangents = _as_tuple(tree_map(lambda x: torch.randn_like(x), primals))
537 expected = _autograd_grad(_as_tuple(wrapped_fn(*args, **kwargs)), diff_args)
/aosp_15_r20/external/pytorch/test/
H A Dtest_decomp.py139 def _as_tuple(val): function
150 _as_tuple(result),
152 _as_tuple(cotangents),
/aosp_15_r20/external/pytorch/torch/_functorch/
H A Deager_transforms.py156 def _as_tuple(val): function
1432 flat_outputs = _as_tuple(output)
H A Dvmap.py82 def _as_tuple( function