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)
|