| /aosp_15_r20/external/pytorch/test/torch_np/numpy_tests/core/ |
| H A D | test_shape_base.py | 26 atleast_1d,
41 atleast_1d,
64 res = [atleast_1d(a), atleast_1d(b)]
71 res = [atleast_1d(a), atleast_1d(b)]
78 res = [atleast_1d(a), atleast_1d(b)]
87 res = [atleast_1d(a), atleast_1d(b)]
93 assert atleast_1d(3).shape == (1,)
94 assert atleast_1d(3j).shape == (1,)
95 assert atleast_1d(3.0).shape == (1,)
96 assert atleast_1d([[2, 3], [4, 5]]).shape == (2, 2)
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorTransformations.cpp | 170 Tensor atleast_1d(const Tensor& self) { in atleast_1d() function
179 std::vector<Tensor> atleast_1d(TensorList tensors) { in atleast_1d() function
182 return at::native::atleast_1d(input); in atleast_1d()
|
| H A D | TensorShape.cpp | 2815 auto rep = at::atleast_1d(tensors); in hstack()
2825 auto rep = at::atleast_1d(tensors); in hstack_out()
|
| /aosp_15_r20/external/pytorch/test/distributed/_tensor/ |
| H A D | test_view_ops.py | 197 self.dimmap_test(torch.atleast_1d, (randn(()),), (Singleton(),))
198 self.dimmap_test(torch.atleast_1d, (randn(24),), (InputDim(0),))
199 self.dimmap_test(torch.atleast_1d, (randn(24, 36),), (InputDim(0), InputDim(1)))
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | functional.py | 1487 def atleast_1d(*tensors): function
1517 return handle_torch_function(atleast_1d, tensors, *tensors)
1520 return _VF.atleast_1d(tensors) # type: ignore[attr-defined]
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/utils/ |
| H A D | np_utils.py | 93 l2 = np.atleast_1d(np.linalg.norm(x, order, axis))
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesDecompositions.cpp | 73 OP_DECOMPOSE(atleast_1d); in TORCH_LIBRARY_IMPL()
74 OP_DECOMPOSE2(atleast_1d, Sequence); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_schema_check.py | 344 expected = torch.atleast_1d([])
346 actual = torch.atleast_1d([])
|
| H A D | test_view_ops.py | 1568 self._test_atleast_dim(torch.atleast_1d, np.atleast_1d, device, dtype)
1596 self._test_atleast(device, torch.atleast_1d)
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| H A D | np_array_ops.py | 123 shape = atleast_1d(shape)
1055 arrays = [atleast_1d(a) for a in tup]
1125 def atleast_1d(*arys): function
1153 a = atleast_1d(a)
|
| H A D | np_math_ops.py | 1220 a_t = np_array_ops.atleast_1d(a)
|
| /aosp_15_r20/external/pytorch/test/export/ |
| H A D | testing.py | 68 aten.atleast_1d.Sequence,
|
| /aosp_15_r20/external/pytorch/functorch/op_analysis/ |
| H A D | public_api | 165 atleast_1d
|
| H A D | annotated_ops | 44 atleast_1d, view/reshape
|
| /aosp_15_r20/external/pytorch/torch/_numpy/ |
| H A D | _funcs_impl.py | 50 def atleast_1d(*arys: ArrayLike): function
51 res = torch.atleast_1d(*arys)
1129 t_a, t_b = torch.atleast_1d(a, b)
|
| /aosp_15_r20/external/pytorch/docs/source/ |
| H A D | masked.rst | 273 atleast_1d
|
| H A D | torch.rst | 518 atleast_1d
|
| /aosp_15_r20/external/pytorch/test/torch_np/ |
| H A D | test_basic.py | 270 seq_funcs = [w.atleast_1d, w.atleast_2d, w.atleast_3d, w.broadcast_arrays]
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset11.py | 1309 def atleast_1d(g: jit_utils.GraphContext, self: torch._C.Value): function
1434 tensor_list = atleast_1d(g, tensor_list)
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | math_ops.py | 319 torch.atleast_1d(a),
|
| /aosp_15_r20/external/pytorch/torch/distributed/tensor/_ops/ |
| H A D | _view_ops.py | 458 torch.atleast_1d: lambda x: dim_pad_left(x.ndim, 1),
|
| /aosp_15_r20/external/pytorch/torch/_subclasses/ |
| H A D | functional_tensor.py | 104 torch.ops.aten.atleast_1d.default, # type: ignore[has-type]
|
| /aosp_15_r20/external/pytorch/test/torch_np/numpy_tests/linalg/ |
| H A D | test_linalg.py | 932 s = np.atleast_1d(s)
933 ld = np.atleast_1d(ld)
|
| /aosp_15_r20/external/tensorflow/third_party/py/numpy/tf_numpy_api/ |
| H A D | tensorflow.experimental.numpy.pbtxt | 224 name: "atleast_1d"
|
| /aosp_15_r20/external/pytorch/torch/_refs/ |
| H A D | __init__.py | 2614 def atleast_1d( function
2647 unsqueeze_atleast_1d = partial(_unsqueeze_atleast, atleast_1d, 0)
3922 aligned_tensors = atleast_1d(*tensors)
|