Home
last modified time | relevance | path

Searched refs:to_numpy (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcustom_op_db.py29 def to_numpy(tensor): function
34 x_np = to_numpy(x)
62 return torch.tensor(to_numpy(x) * to_numpy(y), device=x.device)
92 return torch.tensor(to_numpy(x) * scalar, device=x.device)
119 x = to_numpy(x)
157 x = to_numpy(x)
158 ind = to_numpy(ind)
203 x_np = to_numpy(x)
234 return torch.tensor(np.copy(to_numpy(x).reshape(shape)), device=x.device)
268 np_xs = [to_numpy(x) for x in xs]
[all …]
H A Dautograd_function_db.py25 def to_numpy(tensor): function
32 input_np = to_numpy(input)
89 input_np = to_numpy(input)
108 return torch.tensor(to_numpy(x) * to_numpy(y), device=x.device)
181 x_np = to_numpy(x)
213 x = to_numpy(x)
298 x = to_numpy(x)
299 ind = to_numpy(ind)
/aosp_15_r20/external/pytorch/docs/source/notes/
H A Dextending.func.rst76 def to_numpy(tensor):
84 x = to_numpy(x)
137 x = to_numpy(x)
138 ind = to_numpy(ind)
362 def to_numpy(tensor):
369 x = to_numpy(x)
432 x = to_numpy(x)
433 ind = to_numpy(ind)
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/summary_ops/
H A Dsummary_ops_test.py62 self.assertEqual(42, to_numpy(value))
79 self.assertEqual(42, to_numpy(value))
111 self.assertAllEqual([[1, 2], [3, 4]], to_numpy(value))
122 self.assertAllEqual(expected, to_numpy(value))
137 self.assertAllEqual(expected, to_numpy(value))
146 self.assertAllEqual([b'foo', b'bar'], to_numpy(value))
161 self.assertEqual(42, to_numpy(events[1].summary.value[0]))
1625 def to_numpy(summary_value): function
/aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/
H A Dcore.py294 def to_numpy(t): function
306 return self.transform(to_numpy)
/aosp_15_r20/external/pytorch/test/
H A Dtest_tensor_creation_ops.py3828 def to_numpy(tensor): function
3831 return memoryview(to_numpy(tensor))
3906 self._test_alias_with_cvt(to_numpy, device, dtype)
3955 self._test_copy_with_cvt(to_numpy, device, dtype)
4106 (to_numpy(original), {}),
H A Dtest_reductions.py3584 def to_numpy(input): function
3596 expected = op.ref(to_numpy(t), *sample_input.args,
/aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/
H A Ddataset_ops.py4635 def to_numpy(x): function
4643 return nest.map_structure(to_numpy, next(self._iterator))
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_eager_transforms.py1350 input_np = to_numpy(input) # noqa: F821
1372 input_np = to_numpy(input) # noqa: F821