Searched refs:numpy_op (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_binary_ufuncs.py | 2155 for torch_op, alias, numpy_op in ops:
2163 numpy_result = numpy_op(a_np, b_np)
2189 for torch_op, alias, numpy_op in ops:
2190 numpy_result = numpy_op(a_np, b_np)
2242 for torch_op, alias, numpy_op in ops:
2243 numpy_result = numpy_op(a_np, b_np)
2312 for torch_op, numpy_op in ops:
2316 numpy_result_1 = numpy_op(a_np, b_np)
2318 numpy_result_2 = numpy_op(b_np, a_np)
3144 for torch_op, numpy_op in op_pairs:
[all …]
|
| H A D | test_reductions.py | 797 def _test_dim_ops(self, pytorch_op, numpy_op, argument
807 expected = numpy_op(tensor.cpu().numpy(), dim)
2503 numpy_op = np.median if op == torch.median else np.nanmedian
2514 self.assertEqual(res.item(), numpy_op(t.cpu().numpy()))
2529 ref = numpy_op(t_numpy, dim, keepdims=True)[mask.cpu().numpy()]
2608 numpy_op = getattr(np, op)
2615 expected = numpy_op(a.cpu().numpy(), q.cpu().numpy(), dim,
2739 numpy_op = np.var
2742 numpy_op = np.std
2746 numpy_result = numpy_op(a, **numpy_kwargs)
|