Searched refs:argmin (Results 1 – 6 of 6) sorted by relevance
149 tensor_argmin = tensor.argmin(); in test_simple_argmin()155 tensor_argmin = tensor.argmin(); in test_simple_argmin()243 tensor_argmin = tensor.argmin(dim); in test_argmin_dim()265 tensor_argmin = tensor.argmin(dim); in test_argmin_dim()
54 gpu_out_min.device(gpu_device) = gpu_in.argmin(); in test_cuda_simple_argmax()203 gpu_out.device(gpu_device) = gpu_in.argmin(dim); in test_cuda_argmin_dim()231 gpu_out.device(gpu_device) = gpu_in.argmin(dim); in test_cuda_argmin_dim()
130 assertThat(Utils.argmin(new double[]{5, 2, 1, -10, -20, 5, 19, 100}), is(4)); in testArgmin()135 assertThat(Utils.argmin(new double[]{}), is(0)); in testArgmin_empty()
69 best_shift0 = shifts[numpy.argmin(residuals0)]70 best_shift1 = shifts[numpy.argmin(residuals1)]
124 public static int argmin(double[] a) { in argmin() method in Utils156 bestShift = argmin(stddevs) * shiftStep + bestShift - shiftStep * steps / 2; in findBestShift()
627 argmin() const { in argmin() function653 argmin(const int return_dim) const { in argmin() function